pytorch预训练模型下载保存路径更改

正常情况下，torch预训练模型在线下载，下模型后的地址默认是：

~/.cache/torch/hub/checkpoints

如果没有预先下载好预训练模型，在运行这个代码后，自动下载预训练模型的。

如果要更改路径，有两种办法：

两种办法

1. 在from_pretrained 函数中，指定cache_dir，如：

pipeline = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16, cache_dir="./models/")

2. 修改环境变量（TORCH_HOME或者HF_HOME）

• PyTorch：设置 TORCH_HOME 环境变量到目标路径
• Hugging Face：设置 HF_HOME 或 TRANSFORMERS_CACHE 环境变量到新路径

import os
os.environ['TORCH_HOME']='./torch-models/'