Spaces:
Running
Running
[fix] option field rope in architecture model.
Browse files- app/models.py +2 -2
app/models.py
CHANGED
@@ -37,8 +37,8 @@ class Architecture(BaseModel):
|
|
37 |
attention_key_length: int = Field(alias="attentionKeyLength")
|
38 |
attention_value_length: int = Field(alias="attentionValueLength")
|
39 |
attention_causal: bool = Field(alias="attentionCausal")
|
40 |
-
rope_dimension_count: int = Field(alias="ropeDimensionCount")
|
41 |
-
rope_frequency_base: int = Field(alias="ropeFrequencyBase")
|
42 |
vocabulary_length: int = Field(alias="vocabularyLength")
|
43 |
embedding_gqa: int = Field(alias="embeddingGQA")
|
44 |
embedding_key_gqa: int = Field(alias="embeddingKeyGQA")
|
|
|
37 |
attention_key_length: int = Field(alias="attentionKeyLength")
|
38 |
attention_value_length: int = Field(alias="attentionValueLength")
|
39 |
attention_causal: bool = Field(alias="attentionCausal")
|
40 |
+
rope_dimension_count: int = Field(None, alias="ropeDimensionCount")
|
41 |
+
rope_frequency_base: int = Field(None, alias="ropeFrequencyBase")
|
42 |
vocabulary_length: int = Field(alias="vocabularyLength")
|
43 |
embedding_gqa: int = Field(alias="embeddingGQA")
|
44 |
embedding_key_gqa: int = Field(alias="embeddingKeyGQA")
|