DiffFashion论文复现

论文地址：https://arxiv.org/abs/2302.06826

论文git：GitHub - Rem105-210/DiffFashion

遇到的问题：

安装torch版本的时候：

pip install torch==1.13.1+cu111

ERROR：No matching distribution found for torch==1.13.1+cu111

换成 1.13.1+cu116 之后，运行代码的时候，提示无GPU：

重新卸载环境，然后安装，结果发现磁盘空间不够了= =，难以想象，服务器只有1T的硬盘

改为使用 ldm 环境：

修改代码：

image_editor.py

class ImageEditor:
    def __init__(self, args) -> None:
        # ...将下述代码改一下
        # self.device = torch.device(
        #     f"cuda:{self.args.gpu_id}" if torch.cuda.is_available() else "cpu"
        # )
        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

可以正常执行了：