AMAAI-Lab / mustango

Mustango: Toward Controllable Text-to-Music Generation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

this is no 'pytorch_model_main.bin' in mustango of huggingface.but in code train.py row395 : model.load_state_dict(torch.load("{}/pytorch_model_main.bin".format(hf_model_path), map_location="cpu"))

bocaidoufuyushiren opened this issue · comments

this is no 'pytorch_model_main.bin' in mstango of huggingface.but in code train.py row395 : model.load_state_dict(torch.load("{}/pytorch_model_main.bin".format(hf_model_path), map_location="cpu"))
image
image
how can I finetune this model?

hi, did you resolve this issue?