Spaces:

hyz317
/

StdGEN

Running on L40S

YulianSa commited on 15 days ago

Commit

9516700

1 Parent(s): 405fc9c

update

Files changed (1) hide show

infer_api.py CHANGED Viewed

@@ -283,8 +283,8 @@ def save_image_numpy(ndarr):
 @spaces.GPU
 def run_multiview_infer(data, pipeline, cfg: TestConfig, num_levels=3):
-    infer_multiview_pipeline.unet.enable_xformers_memory_efficient_attention()
     if cfg.seed is None:
         generator = None
     else:

 @spaces.GPU
 def run_multiview_infer(data, pipeline, cfg: TestConfig, num_levels=3):
+    pipeline.unet.enable_xformers_memory_efficient_attention()
     if cfg.seed is None:
         generator = None
     else: