From f3daa5bdebb4e4c7adfdebc3d9d7f23c59ef75fb Mon Sep 17 00:00:00 2001 From: aiXander Date: Thu, 15 Aug 2024 04:49:24 -0700 Subject: [PATCH] ready for cog push --- main.py | 2 +- trainer/config.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) 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"