test t5

stable/stable_audio_tools/models/conditioners.py

@@ -5,7 +5,7 @@ import logging, warnings
 import string
 import typing as tp
 import gc
-
+import os
 from .adp import NumberEmbedder
 from ..inference.utils import set_audio_channels
 from .factory import create_pretransform_from_config
@@ -283,6 +283,8 @@ 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()
+                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)
                 model.load_state_dict(ckpt,strict=True)