phate334 commited on
Commit
c0976bd
·
1 Parent(s): e20ecc6

[fix] option field rope in architecture model.

Browse files
Files changed (1) hide show
  1. 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")