SD35-IP-Adapter

Running on L40S

multimodalart HF staff commited on 7 days ago

Commit

e665eb0

•

1 Parent(s): 29282b8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ pipe = StableDiffusion3Pipeline.from_pretrained(
     model_path,
     transformer=transformer,
     torch_dtype=torch.bfloat16
-)
 pipe.init_ipadapter(
     ip_adapter_path=ipadapter_path,
@@ -58,7 +58,7 @@ def process_image(
     height,
     progress=gr.Progress(track_tqdm=True),
 ):
-    pipe.to("cuda")
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)

     model_path,
     transformer=transformer,
     torch_dtype=torch.bfloat16
+).to("cuda")
 pipe.init_ipadapter(
     ip_adapter_path=ipadapter_path,
     height,
     progress=gr.Progress(track_tqdm=True),
 ):
+    #pipe.to("cuda")
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)