Fix bug for p-tuning

xusenlin · xusenlin · commit 1801f3905792 · 2023-12-07T14:09:34.000+08:00
diff --git a/api/config.py b/api/config.py
@@ -117,7 +117,8 @@ class Settings(BaseModel):
         description="Whether to load the model using ptuning_v2."
     )
     pre_seq_len: Optional[bool] = Field(
-        default=get_bool_env("PRE_SEQ_LEN"),
+        default=int(get_env("PRE_SEQ_LEN", 128)),
+        ge=0,
         description="PRE_SEQ_LEN for ptuning_v2."
     )
 

Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,8 @@ class Settings(BaseModel):`
`117`	`117`	`description="Whether to load the model using ptuning_v2."`
`118`	`118`	`)`
`119`	`119`	`pre_seq_len: Optional[bool] = Field(`
`120`		`- default=get_bool_env("PRE_SEQ_LEN"),`
	`120`	`+ default=int(get_env("PRE_SEQ_LEN", 128)),`
	`121`	`+ ge=0,`
`121`	`122`	`description="PRE_SEQ_LEN for ptuning_v2."`
`122`	`123`	`)`
`123`	`124`