new

pipeline/kiss3d_wrapper.py

@@ -97,7 +97,7 @@ def init_wrapper_from_config(config_path):
 
         flux_redux_pipe.to(device=flux_device)
 
-    logger.warning(f"GPU memory allocated after load flux model on {flux_device}: {torch.cuda.memory_allocated(device=flux_device) / 1024**3} GB")
+    # logger.warning(f"GPU memory allocated after load flux model on {flux_device}: {torch.cuda.memory_allocated(device=flux_device) / 1024**3} GB")
 
     # TODO: load pulid model
 
@@ -147,7 +147,7 @@ def init_wrapper_from_config(config_path):
     recon_model.load_state_dict(state_dict, strict=True)
     recon_model.to(recon_device)
     recon_model.eval()
-    logger.warning(f"GPU memory allocated after load reconstruction model on {recon_device}: {torch.cuda.memory_allocated(device=recon_device) / 1024**3} GB")
+    # logger.warning(f"GPU memory allocated after load reconstruction model on {recon_device}: {torch.cuda.memory_allocated(device=recon_device) / 1024**3} GB")
 
     # load llm
     llm_configs = config_.get('llm', None)
@@ -156,7 +156,7 @@ def init_wrapper_from_config(config_path):
         llm_device = llm_configs.get('device', 'cpu')
         llm, llm_tokenizer = load_llm_model(llm_configs['base_model'])
         llm.to(llm_device)
-        logger.warning(f"GPU memory allocated after load llm model on {llm_device}: {torch.cuda.memory_allocated(device=llm_device) / 1024**3} GB")
+        # logger.warning(f"GPU memory allocated after load llm model on {llm_device}: {torch.cuda.memory_allocated(device=llm_device) / 1024**3} GB")
     else:
         llm, llm_tokenizer = None, None
 
@@ -462,7 +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,