new

pipeline/kiss3d_wrapper.py

@@ -74,8 +74,9 @@ def init_wrapper_from_config(config_path):
         flux_pipe = FluxImg2ImgPipeline.from_single_file(flux_base_model_pth, torch_dtype=dtype_[flux_dtype], token=access_token)
     else:
         flux_pipe = FluxImg2ImgPipeline.from_pretrained(flux_base_model_pth, torch_dtype=dtype_[flux_dtype], token=access_token)
-    flux_pipe.enable_vae_slicing()
-    flux_pipe.enable_xformers_memory_efficient_attention()
+    # flux_pipe.enable_vae_slicing()
+    # flux_pipe.enable_xformers_memory_efficient_attention()
+    flux_pipe.enable_sequential_cpu_offload()
     # load flux model and controlnet
     if flux_controlnet_pth is not None and False:
         flux_controlnet = FluxControlNetModel.from_pretrained(flux_controlnet_pth, torch_dtype=torch.bfloat16)