File size: 652 Bytes
72895aa
 
 
 
 
 
 
 
 
 
 
 
 
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 = "xianbao/demo_test"

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