项目页面 | 论文 | Huggingface

Perp-Neg Stable Diffusion

这是使用Perp-Neg采样与Stable Diffusion模型的代码仓库，如重新构想负面提示词算法：将2D扩散转化为3D，缓解双面问题等中所述。

运行代码

运行扩散模型的组合式能量模型

python scripts/image_sample_stable_diffusion.py --prompt="戴墨镜的男孩|黑框墨镜" --weights="1|-1.5" --pipeline='cebm' --num_images=1

运行PerpNeg

python scripts/image_sample_stable_diffusion.py --prompt="戴墨镜的男孩|黑框墨镜" --weights="1|-1.5" --pipeline='perpneg' --num_images=1

运行PerpNeg旋转

python scripts/image_sample_stable_diffusion.py --prompt="正面视角的狮子照片|侧面视角的狮子照片" --weights="1|-1.5|-1.5" --pipeline='rotation_perpneg' --num_images=1

PerpNeg-Stable DreamFusion

请使用stable-dreamfusion/scripts/run_if2_perpneg.sh中的脚本运行Stable DreamFusion + PerpNeg以避免双面问题。

该代码主要基于Stable-DreamFusion

如果您发现它对您的工作有用，请引用：

@article{armandpour2023re,
  title={Re-imagine the Negative Prompt Algorithm: Transform 2D Diffusion into 3D, alleviate Janus problem and Beyond},
  author={Armandpour, Mohammadreza and Zheng, Huangjie and Sadeghian, Ali and Sadeghian, Amir and Zhou, Mingyuan},
  journal={arXiv preprint arXiv:2304.04968},
  year={2023}
}