OminiControl-duplicated

Running on Zero

Yuanshi commited on Dec 6, 2024

Commit

9ce9f52

verified ·

1 Parent(s): 1a1d307

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",