Kolors-IPAdapter

Running on Zero

multimodalart HF staff commited on Jul 18

Commit

7f31109

•

1 Parent(s): 9d21a93

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def infer(prompt, ip_adapter_image, negative_prompt="", seed=42, randomize_seed=
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     pipe.set_ip_adapter_scale([ip_adapter_scale])
     image = pipe(

         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
+    pipe = pipe.to("cuda")
     pipe.set_ip_adapter_scale([ip_adapter_scale])
     image = pipe(