Local model problem
pendekarcode opened this issue · comments
pendekarcode commented
I downloaded the model and saved it on a local PC. By code:
namefolder= 'mrm8488-t5-base'
tokenizer = AutoTokenizer.from_pretrained('mrm8488/t5-base-finetuned-common_gen')
model = AutoModelForSeq2SeqLM.from_pretrained('mrm8488/t5-base-finetuned-common_gen')
tokenizer.save_pretrained("./"+namefolder)
model.save_pretrained("./"+namefolder)
I can not run the model. Error:
site-packages\keytotext\pipeline.py", line 75, in pipeline
task, list(SUPPORTED_TASKS.keys())
KeyError: "Unknown task E:\PH\_MODEL_TRANSFORMERS\mrm8488-t5-base, available tasks are ['k2t', 'k2t-base', 'mrm8488/t5-base-finetuned-common_gen', 'k2t- new']"
Thanks