Spaces:

jiuface
/

realflux

Running on Zero

jiuface commited on 9 days ago

Commit

c911174

•

1 Parent(s): a3caeb6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ login(token=HF_TOKEN)
 # Initialize the base model and specific LoRA
 model_url = "https://huggingface.co/SG161222/RealFlux_1.0b_Dev/resolve/main/1%20-%20Compact%20Version/RealFlux_1.0b_Dev_Compact.safetensors"
-pipe = DiffusionPipeline.from_single_file(model_url, torch_dtype=torch.bfloat16)
 pipe.to("cuda")

 # Initialize the base model and specific LoRA
 model_url = "https://huggingface.co/SG161222/RealFlux_1.0b_Dev/resolve/main/1%20-%20Compact%20Version/RealFlux_1.0b_Dev_Compact.safetensors"
+pipe = DiffusionPipeline.from_pretrained("SG161222/RealFlux_1.0b_Dev", torch_dtype=torch.bfloat16)
 pipe.to("cuda")