huggingface 国内如何连接

cache_dir='E:\\huggingface_model_cache\\'
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
os.environ["HF_DATASETS_CACHE"] = cache_dir
os.environ["HF_HOME"] = cache_dir
os.environ["HUGGINGFACE_HUB_CACHE"] = cache_dir
os.environ["TRANSFORMERS_CACHE"] = cache_dir




from transformers import pipeline
import os

pipe = pipeline('sentiment-analysis')
print(pipe('今天真高兴'))
 
刚刚测试了,高版本的transformers的4.31.0能这么用,但是低版本的不行哦。 建议升级到最新的版本
posted @ 2024-01-28 08:46  纯丿乱  阅读(316)  评论(0编辑  收藏  举报