t5

stable/stable_audio_tools/models/conditioners.py

@@ -283,9 +283,9 @@ class T5Conditioner(Conditioner):
                 # self.tokenizer = T5Tokenizer.from_pretrained(t5_model_name, model_max_length = max_length)
                 # model = T5EncoderModel.from_pretrained(t5_model_name, max_length=max_length).train(enable_grad).requires_grad_(enable_grad)
                 self.tokenizer = AutoTokenizer.from_pretrained(t5_model_name)
-                cwd = os.listdir('./')
-                print("==========", cwd)
-                ckpt = torch.load('../try_t5.pt')
+                # cwd = os.listdir('./')
+                # print("==========", cwd)
+                ckpt = torch.load('./stable/stable_audio_tools/try_t5.pt')
                 model = T5EncoderModel.from_pretrained(t5_model_name).train(enable_grad).requires_grad_(enable_grad).to(torch.float16)
                 model.load_state_dict(ckpt,strict=True)