Prakh24s commited on
Commit
b6919f9
·
verified ·
1 Parent(s): e20d29f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- ).to
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