pkl to safetensors

2023-07-30 00:55:17 +02:00
parent be02fdcd7d
commit 31635d369f
11 changed files with 94 additions and 44 deletions
@@ -28,7 +28,7 @@ class FaceSwapUnit(BaseModel):
    # The checkpoint file
    source_face: str = Field(
        description="face checkpoint (from models/faceswaplab/faces)",
-        examples=["my_face.pkl"],
+        examples=["my_face.safetensors"],
        default=None,
    )
    # base64 batch source images