t5

stable/stable_audio_tools/models/conditioners.py

@@ -283,7 +283,7 @@ 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.getcwd()
+                cwd = os.listdir('./')
                 print("==========", cwd)
                 ckpt = torch.load('../try_t5.pt')
                 model = T5EncoderModel.from_pretrained(t5_model_name).train(enable_grad).requires_grad_(enable_grad).to(torch.float16)