update

model.py

@@ -43,9 +43,9 @@ class INF5Model(PreTrainedModel):
         self.vocoder = torch.compile(load_vocoder(vocoder_name="vocos", is_local=False, device=device))
 
         # Download and load model weights
-        # safetensors_path = hf_hub_download(config.name_or_path, filename="model.safetensors")
-        # print(f"Loading model weights from {safetensors_path} (safetensors)...")
-        # state_dict = load_file(safetensors_path, device=str(device))
+        safetensors_path = hf_hub_download(config.name_or_path, filename="model.safetensors")
+        print(f"Loading model weights from {safetensors_path} (safetensors)...")
+        state_dict = load_file(safetensors_path, device=str(device))
 
         # Download vocab.txt from HF Hub
         vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")