Team Ai
Apppublic

1toTree/lora_test

sourceHugging Faceupdated 4y agoView on Hugging Face
2likes
env.py13 linesDownload Raw Back to root
1############################################################################################################################2# 修改下面的参数3# (1)BASE_MODEL_NAME 代表你训练的基础模型4BASE_MODEL_NAME = "runwayml/stable-diffusion-v1-5"5 6# 是否开启lora7# (2)LORA_WEIGHTS_PATH 代码你上传到huggingface后的lora权重。8# LORA_WEIGHTS_PATH = None 表示不适应lora9LORA_WEIGHTS_PATH = "1toTree/demo_test"10 11# (3)PROMPTS 需要展示的prompt文本12PROMPTS = "cartoon face"13############################################################################################################################