new

pipeline/kiss3d_wrapper.py

@@ -69,7 +69,7 @@ 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)
+    taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype_[flux_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: