diff --git a/main.py b/main.py index 49b004f..64a134a 100755 --- a/main.py +++ b/main.py @@ -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 diff --git a/trainer/config.py b/trainer/config.py index db18c59..40aa4af 100644 --- a/trainer/config.py +++ b/trainer/config.py @@ -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"