new

pipeline/kiss3d_wrapper.py

@@ -161,7 +161,7 @@ def init_wrapper_from_config(config_path):
         # 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
-
+    torch.cuda.empty_cache()
     return kiss3d_wrapper(
         config = config_,
         flux_pipeline = flux_pipe,