Spaces:

HorizonRobotics
/

EmbodiedGen-Text-to-3D

Running on Zero

xinjie.wang commited on Jun 12

Commit

50772e1

1 Parent(s): 3feae4a

update

Files changed (1) hide show

common.py CHANGED Viewed

@@ -695,7 +695,7 @@ def text2image_fn(
     if ip_image is not None:
         pipeline.set_ip_adapter_scale([ip_adapt_scale])
-    pipeline.cuda()
     images = text2img_gen(
         prompt=prompt,
         n_sample=n_sample,

     if ip_image is not None:
         pipeline.set_ip_adapter_scale([ip_adapt_scale])
+    pipeline = pipeline.to("cuda")
     images = text2img_gen(
         prompt=prompt,
         n_sample=n_sample,