lora_test2 / env.py
junnyu's picture
Upload with huggingface_hub
8c1c4dc
############################################################################################################################
# 修改下面的参数
# (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"
############################################################################################################################