Spaces:

yasserrmd
/

MagicDoodles

Running on Zero

yasserrmd commited on Dec 16, 2024

Commit

e2bd59b

verified ·

1 Parent(s): 9388d07

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ device=torch.device('cuda')
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 print(prompt)
-if style='shou_xin':
     prompt= "shou_xin, " + prompt
     pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
 else:

 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 print(prompt)
+if style=='shou_xin':
     prompt= "shou_xin, " + prompt
     pipe.load_lora_weights("Datou1111/shou_xin", weight_name="shou_xin.safetensors")
 else: