1. 拉取代码

git clone https://github.com/TEXTurePaper/TEXTurePaper.git

2. 准备环境所需要的包

在这里插入图片描述

3. 由于代码自动下载模型,容易中断报错,所以采取将模型下载到本地,然后修改代码默认的路径为本地的模型路径。

# 安装 tree(如果没有)
sudo apt-get update && sudo apt-get install tree -y

# 显示 /root/autodl-tmp 下的目录结构
tree -L 3 /root/autodl-tmp

在这里插入图片描述
在这里插入图片描述

4. 将以上模型在本地准备好,修改代码默认的路径

TEXTurePaper-main/src/configs/train_config.py中的48行,将stabilityai/stable-diffusion-2-depth改为本地的/root/autodl-tmp/stabilityaistable-diffusion-2-depth,根据自己的路径修改
Code/TEXTurePaper-main/src/stable_diffusion_depth.py60行的stabilityai/stable-diffusion-2-inpainting改为/root/autodl-tmp/stable-diffusion-2-inpainting,根据自己的路径修改

5. 激活环境,进入项目目录并运行指令

source activate autodl-tmp/envs
cd /root/Code/TEXTurePaper-main
python -m scripts.run_texture --config_path=configs/text_guided/napoleon.yaml

运行的示意图:
在这里插入图片描述

以上内容可以总结为


【保姆级教程】TEXTurePaper运行环境搭建与Stable Diffusion模型本地化

在跑 TEXTurePaper 项目的时候,很多同学会遇到一个坑:
默认配置会直接从 HuggingFace 下载 stabilityai/stable-diffusion-2-depth 模型,网络环境不好时经常中断,训练根本跑不通。这里分享一下我的解决方案。


一、模型准备

在源码 Code/TEXTurePaper-main/src/configs/train_config.py48 行,默认配置如下:

diffusion_name: str = 'stabilityai/stable-diffusion-2-depth'

⚠️ 这样写会强制从官网拉取模型。为避免下载中断,建议手动下载到本地:

  1. 去 HuggingFace 官网下载 stabilityai/stable-diffusion-2-depth

  2. 上传到服务器(推荐使用 FileZilla 拖拽上传,或者 autodl 自带上传功能)。

  3. 上传路径建议:

    /root/autodl-tmp/stable-diffusion-2-depth
    

保持文件夹结构与官网一致,这样调用时不会出错。


二、修改代码路径

修改以下两个文件,把默认的 HuggingFace 名称替换为本地路径:

  • stable_diffusion_depth.py 第 60 行

    diffusion_name: str = '/root/autodl-tmp/stable-diffusion-2-depth'
    
  • train_config.py 第 48 行

    diffusion_name: str = '/root/autodl-tmp/stable-diffusion-2-depth'
    

这样,模型就会直接从本地读取,而不是联网下载。


三、环境激活

进入虚拟环境:

source activate /root/autodl-tmp/envs

四、运行命令

在项目根目录下执行:

python -m scripts.run_texture --config_path=configs/text_guided/napoleon.yaml

如果路径和环境没问题,程序就能正常跑起来。


五、总结

  • 踩坑点:默认配置会强制联网下载,容易失败。
  • 解决办法:提前下载好模型 → 上传服务器 → 修改路径。
  • 关键代码位置train_config.py(48行)和 stable_diffusion_depth.py(60行)。

这样处理后,TEXTurePaper 在国内环境也能稳定运行。


有问题可以在评论区留言,或者直接私信我交流。


Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐