new

pipeline/kiss3d_wrapper.py

@@ -67,7 +67,7 @@ def init_wrapper_from_config(config_path):
     flux_dtype = config_['flux'].get('dtype', 'bf16')
     flux_controlnet_pth = config_['flux'].get('controlnet', None)
     # flux_lora_pth = config_['flux'].get('lora', None)
-    flux_lora_pth = hf_hub_download(repo_id="LTT/xxx-ckpt", filename="rgb_normal_large.safetensors", repo_type="model", token=access_token)
+    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)
     
     if flux_base_model_pth.endswith('safetensors'):
@@ -114,7 +114,7 @@ def init_wrapper_from_config(config_path):
     )
     
     # unet_ckpt_path = config_['multiview'].get('unet', None)
-    unet_ckpt_path = hf_hub_download(repo_id="LTT/PRM", filename="flexgen_19w.ckpt", repo_type="model")
+    unet_ckpt_path = hf_hub_download(repo_id="LTT/Kiss3DGen", filename="flexgen_19w.ckpt", repo_type="model", token=access_token)
     if unet_ckpt_path is not None:
         state_dict = torch.load(unet_ckpt_path, map_location='cpu')['state_dict']
         state_dict = {k[10:]: v for k, v in state_dict.items() if k.startswith('unet.unet.')}