File size: 651 Bytes
05654ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
############################################################################################################################
# 修改下面的参数
# （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"
############################################################################################################################ |