OminiControl

Runtime error

Yuanshi commited on Dec 6, 2024

Commit

1a1d307

verified ·

1 Parent(s): d5fc119

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def init_pipeline():
     pipe = FluxPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
     )
-    pipe = pipe.to("cuda")
     pipe.load_lora_weights(
         "Yuanshi/OminiControl",
         weight_name=f"omini/subject_512.safetensors",

     pipe = FluxPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16
     )
+    # pipe = pipe.to("cuda")
     pipe.load_lora_weights(
         "Yuanshi/OminiControl",
         weight_name=f"omini/subject_512.safetensors",