env.py

############################################################################################################################
# 修改下面的参数
# （1）BASE_MODEL_NAME 代表你训练的基础模型
BASE_MODEL_NAME = "runwayml/stable-diffusion-v1-5"

# 是否开启lora
# （2）LORA_WEIGHTS_PATH 代码你上传到huggingface后的lora权重。
# LORA_WEIGHTS_PATH = None 表示不适应lora
LORA_WEIGHTS_PATH = "junnyu/demo_test"

# （3）PROMPTS 需要展示的prompt文本
PROMPTS = "A photo of sks dog in a bucket"
############################################################################################################################