| from utils import upload_lora_folder | |
| upload_dir = "cats" # 我们需要上传的文件夹目录 | |
| repo_name = "CatsSnowforest" # 我们需要上传的repo名称 | |
| pretrained_model_name_or_path = "runwayml/stable-diffusion-v1-5" # 训练该模型所使用的基础模型 | |
| prompt = "a cat looking at the Snow forest" # 搭配该权重需要使用的Prompt文本 | |
| upload_lora_folder( | |
| upload_dir=upload_dir, | |
| repo_name=repo_name, | |
| pretrained_model_name_or_path=pretrained_model_name_or_path, | |
| prompt=prompt, | |
| ) |