nodes/edenartlab-sd-lora-trainer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ready for cog push

Browse Source
This commit is contained in:
aiXander
2024-08-15 04:49:24 -07:00
parent 414d43db2d
commit f3daa5bdeb
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+1 -1
View File
@@ -56,7 +56,7 @@ def train(config: TrainingConfig):
if not config.sample_imgs_lora_scale:
if config.sd_model_version == "sdxl":
config.sample_imgs_lora_scale = 0.7
config.sample_imgs_lora_scale = 0.75
else:
config.sample_imgs_lora_scale = 0.85
trainer/config.py
+1 -2
View File
@@ -25,8 +25,7 @@ class ModelPaths:
model_paths = ModelPaths()
# Default SD model download urls in case no local model is found:
SDXL_URL = "https://edenartlab-lfs.s3.amazonaws.com/comfyui/models2/checkpoints/zavychromaxl_v90.safetensors"
#SDXL_URL = "https://edenartlab-lfs.s3.amazonaws.com/models/checkpoints/Eden_SDXL.safetensors"
SDXL_URL = "https://edenartlab-lfs.s3.amazonaws.com/models/checkpoints/Eden_SDXL.safetensors"
#SDXL_URL = "https://huggingface.co/RunDiffusion/Juggernaut-XL-v6/resolve/main/juggernautXL_version6Rundiffusion.safetensors"
SD15_URL = "https://huggingface.co/KamCastle/jugg/resolve/main/juggernaut_reborn.safetensors"