Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,10 +23,10 @@ base_model="stabilityai/stable-diffusion-xl-base-1.0"
|
|
23 |
# ).to("cuda")
|
24 |
base = DiffusionPipeline.from_pretrained(base_model,
|
25 |
torch_dtype=torch.float16, use_safetensors=True
|
26 |
-
)
|
27 |
-
lora_model_path = "./pytorch_weights.safetensors"
|
28 |
|
29 |
-
base.load_lora_weights(lora_model_path)
|
30 |
|
31 |
def swap_face(source_file, target_file,doFaceEnhancer):
|
32 |
|
|
|
23 |
# ).to("cuda")
|
24 |
base = DiffusionPipeline.from_pretrained(base_model,
|
25 |
torch_dtype=torch.float16, use_safetensors=True
|
26 |
+
)
|
27 |
+
# lora_model_path = "./pytorch_weights.safetensors"
|
28 |
|
29 |
+
# base.load_lora_weights(lora_model_path)
|
30 |
|
31 |
def swap_face(source_file, target_file,doFaceEnhancer):
|
32 |
|