Open haorunze opened 1 year ago
我在运行cli_demo_hf.py时,自动下载了几个checkpoint,我想问这几个下载到哪里去了?感谢
如果你是说checkpoint的下载地址的话:
- huggingface版本:https://huggingface.co/THUDM/visualglm-6b 或者 https://cloud.tsinghua.edu.cn/d/43ffb021ca5f4897b56a/
- sat版本:https://github.com/THUDM/VisualGLM-6B/blob/main/model/visualglm.py#L9
如果你是说加载模型时候的加载路径的话,直接设置本地路径就可以,比如
AutoModel.from_pretrained('your/path/to/visualglm-6b')
我在运行cli_demo_hf.py时,自动下载了几个checkpoint,我想问这几个下载到哪里去了?感谢
linux会自动下载到~/.cache/huggingface
linux会自动下载到
~/.cache/huggingface
请问如何更改这个缓存下载路径
export TRANSFORMERS_CACHE=/blabla/cache/
sat下载模型后,cli_demo.py中是那段代码加载模型啊?
如果你是说checkpoint的下载地址的话:
如果你是说加载模型时候的加载路径的话,直接设置本地路径就可以,比如
AutoModel.from_pretrained('your/path/to/visualglm-6b')