| ############################################################################################################################ | |
| # 修改下面的参数 | |
| # （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" | |
| ############################################################################################################################ |