new

pipeline/kiss3d_wrapper.py

@@ -133,6 +133,7 @@ def init_wrapper_from_config(config_path):
     # caption_processor = AutoProcessor.from_pretrained(config_['caption']['base_model'], trust_remote_code=True)
     # logger.warning(f"GPU memory allocated after load caption model on {caption_device}: {torch.cuda.memory_allocated(device=caption_device) / 1024**3} GB")
     caption_processor = None
+    caption_model = None
 
     # load reconstruction model
     logger.info('==> Loading reconstruction model ...')