Hugging Face's logo

Spaces:
yasserrmd
/
MagicDoodles
Running on Zero

App Files Community
yasserrmd commited on
Commit
b875cd2
·
verified ·
1 Parent(s): 85913ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ device=torch.device('cuda')
9
 
10
  # Load the model and LoRA weights
11
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
12
- pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
 
13
  pipe.fuse_lora(lora_scale=1.5)
14
  pipe.to("cuda")
15
 
 
9
 
10
  # Load the model and LoRA weights
11
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
12
+ #pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch", weight_name="FLUX-dev-lora-children-simple-sketch.safetensors")
13
+ pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
14
  pipe.fuse_lora(lora_scale=1.5)
15
  pipe.to("cuda")
16