new

pipeline/kiss3d_wrapper.py

@@ -69,13 +69,14 @@ def init_wrapper_from_config(config_path):
     # flux_lora_pth = config_['flux'].get('lora', None)
     flux_lora_pth = hf_hub_download(repo_id="LTT/Kiss3DGen", filename="rgb_normal_large.safetensors", repo_type="model", token=access_token)
     flux_redux_pth = config_['flux'].get('redux', None)
-    
+    taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
     if flux_base_model_pth.endswith('safetensors'):
         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_vae_tiling()
+    # flux_pipe.enable_vae_tiling()
+    flux_pipe.vae = taef1
     
     # flux_pipe.enable_sequential_cpu_offload()
     # load flux model and controlnet