我在 Chaquopy android studio 项目中的 python 代码:
import torch as tc
from transformers import GPT2Tokenizer, GPT2Model
def generate_text(txt):
"""
Generate chat
https://huggingface.co/gpt2
"""
#Load Model files
tokenizer = GPT2Tokenizer.from_pretrained('assets/') #This line causing error
model = GPT2Model.from_pretrained('assets/')
#Move moel to GPU if avilable
device = tc.device("cuda" if tc.cuda.is_available() else "cpu")
model.to(device)
encoded_input = tokenizer(txt, return_tensors='pt')
output = model(**encoded_input)
return str(output)
现在它显示以下错误:
E/AndroidRuntime: FATAL EXCEPTION: main
Process: com.example.chaquopy_130application, PID: 4867
com.chaquo.python.PyException: HFValidationError: Repo id must use alphanumeric chars or '-', '_', '.', '--' and '..' are forbidden, '-' and '.' cannot start or end the name, max length is 96: 'assets/'.
at <python>.huggingface_hub.utils._validators.validate_repo_id(_validators.py:164)
at <python>.huggingface_hub.utils._validators._inner_fn(_validators.py:110)
at <python>.huggingface_hub.utils._deprecation.inner_f(_deprecation.py:103)
at <python>.transformers.file_utils.get_list_of_files(file_utils.py:2103)
at <python>.transformers.tokenization_utils_base.get_fast_tokenizer_file(tokenization_utils_base.py:3486)
at <python>.transformers.tokenization_utils_base.from_pretrained(tokenization_utils_base.py:1654)
at <python>.pythonScript.generate_text(pythonScript.py:30)
我已经把124M GPT-2型号的所有文件都放了检查站, 编码器.json, hparams.json, model.ckpt.data-00000-of-00001, 模型.ckpt.索引, 模型.ckpt.元, 词汇表“assets”文件夹内的文件。