updated model.py
facing issue :
Traceback (most recent call last):
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\evaluate_tts.py", line 5, in
model = AutoModel.from_pretrained("AI4Bharat/IndicF5", trust_remote_code=True)
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\transformers\models\auto\auto_factory.py", line 559, in from_pretrained
return model_class.from_pretrained(
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\transformers\modeling_utils.py", line 262, in wrapper
return func(*args, **kwargs)
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\transformers\modeling_utils.py", line 4185, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "C:\Users\ravrao.cache\huggingface\modules\transformers_modules\AI4Bharat\IndicF5\b82d286220e3070e171f4ef4b4bd047b9a447c9a\model.py", line 43, in init
self.vocoder = torch.compile(load_vocoder(vocoder_name="vocos", is_local=False, device=device))
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\torch_init.py", line 1723, in compile
return torch._dynamo.optimize(backend=backend, nopython=fullgraph, dynamic=dynamic, disable=disable)(model)
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\torch_dynamo\eval_frame.py", line 583, in optimize
check_if_dynamo_supported()
File "C:\Users\ravrao\Downloads\IndicF5-main\IndicF5-main\TTS-env1\lib\site-packages\torch_dynamo\eval_frame.py", line 535, in check_if_dynamo_supported
raise RuntimeError("Windows not yet supported for torch.compile")
RuntimeError: Windows not yet supported for torch.compile
so directly loading using load_vocoder
This change is causing the below error now to fix this @Ravitejajupally3
45 # Download and load model weights
46 # safetensors_path = hf_hub_download(config.name_or_path, filename="model.safetensors")
47 # print(f"Loading model weights from {safetensors_path} (safetensors)...")
48 # state_dict = load_file(safetensors_path, device=str(device))
49
50 # Download vocab.txt from HF Hub
51 vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")
---> 53 self.ema_model = torch.compile(load_model(
54 DiT,
55 dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),
56 mel_spec_type="vocos",
57 vocab_file=vocab_path,
58 device=device
59 )
60 )
TypeError: load_model() missing 1 required positional argument: 'ckpt_path'