youtube_spleeter / test.py
Niral Patel
change model name
a6ddb91
raw
history blame
521 Bytes
# from transformers import AutoConfig, AutoModel
# config = AutoConfig.from_pretrained("niral-env/youtube_spleeter")
# print(config)
# model = AutoModel.from_pretrained("niral-env/youtube_spleeter", config=config)
# print(model)
# Load model directly
# from transformers import AutoModel
# model = AutoModel.from_pretrained("niral-env/youtube_spleeter")
# print(model)
from transformers import TFAutoModel
# Loading the TensorFlow model
model = TFAutoModel.from_pretrained("niral-env/youtube_spleeter")
print(model)