make save_safetensors: true the default (#2292)
* make save_safetensors: true the default * revert change to model output check
This commit is contained in:
@@ -528,7 +528,7 @@ class ModelOutputConfig(BaseModel):
|
|||||||
output_dir: str = Field(default="./model-out")
|
output_dir: str = Field(default="./model-out")
|
||||||
hub_model_id: Optional[str] = None
|
hub_model_id: Optional[str] = None
|
||||||
hub_strategy: Optional[str] = None
|
hub_strategy: Optional[str] = None
|
||||||
save_safetensors: Optional[bool] = None
|
save_safetensors: Optional[bool] = True
|
||||||
|
|
||||||
|
|
||||||
class MLFlowConfig(BaseModel):
|
class MLFlowConfig(BaseModel):
|
||||||
|
|||||||
Reference in New Issue
Block a user