lora_test / env.py
xiaozeng's picture
Upload with huggingface_hub
05654ff
raw
history blame contribute delete
651 Bytes
############################################################################################################################
# 修改下面的参数
# (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"
############################################################################################################################