File size: 611 Bytes
1126b94 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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,
) |