new

app.py

@@ -149,7 +149,6 @@ def text_to_detailed(prompt, seed=None):
     # print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
     return k3d_wrapper.get_detailed_prompt(prompt, seed)
 
-@spaces.GPU
 def text_to_image(prompt, seed=None, strength=1.0,lora_scale=1.0, num_inference_steps=30, redux_hparam=None, init_image=None, **kwargs):
     # print(f"Before text_to_image: {torch.cuda.memory_allocated() / 1024**3} GB")
     k3d_wrapper.renew_uuid()

pipeline/kiss3d_wrapper.py

@@ -462,6 +462,7 @@ class kiss3d_wrapper(object):
             logger.info(f'Save image to {save_path}')
     
         return preprocessed
+        
     @spaces.GPU
     def generate_3d_bundle_image_text(self, 
                                       prompt,

pipeline/pipeline_config/default.yaml

@@ -6,7 +6,7 @@ flux:
   redux: "black-forest-labs/FLUX.1-Redux-dev"
   num_inference_steps: 20
   seed: 42
-  device: 'cuda:0'
+  device: 'cuda'
 
 multiview:
   base_model: "sudo-ai/zero123plus-v1.2"
@@ -14,20 +14,20 @@ multiview:
   unet: "./checkpoint/zero123++/flexgen_19w.ckpt"
   num_inference_steps: 50
   seed: 42
-  device: 'cuda:0'
+  device: 'cuda'
 
 reconstruction:
   model_config: "./models/lrm/config/PRM_inference.yaml"
   base_model: "./checkpoint/lrm/final_ckpt.ckpt"
-  device: 'cuda:0'
+  device: 'cuda'
 
 caption:
   base_model: "multimodalart/Florence-2-large-no-flash-attn"
-  device: 'cuda:1'
+  device: 'cuda'
 
 llm:
   base_model: "Qwen/Qwen2-7B-Instruct"
-  device: 'cuda:1'
+  device: 'cuda'
 
 use_zero_gpu: false # for huggingface demo only
 3d_bundle_templates: './init_3d_Bundle'