This commit is contained in:
游雁 2024-07-18 13:48:51 +08:00
parent b03c8a5c35
commit 340b6efef2

View File

@ -25,7 +25,7 @@ class ModelDimensions:
n_text_state: int
n_text_head: int
n_text_layer: int
att_type: str
att_type: str = "default"
# class LayerNorm(nn.LayerNorm):