new

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'
+  device: 'cuda:0'
 
 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'
+  device: 'cuda:1'
 
 reconstruction:
   model_config: "./models/lrm/config/PRM_inference.yaml"
   base_model: "./checkpoint/lrm/final_ckpt.ckpt"
-  device: 'cuda'
+  device: 'cuda:1'
 
 caption:
   base_model: "multimodalart/Florence-2-large-no-flash-attn"
-  device: 'cuda'
+  device: 'cuda:1'
 
 llm:
   base_model: "Qwen/Qwen2-7B-Instruct"
-  device: 'cuda'
+  device: 'cuda:1'
 
 use_zero_gpu: false # for huggingface demo only
 3d_bundle_templates: './init_3d_Bundle'