update training workflow

ComfyUI_workflows/traininig_workflow.json

@@ -1,6 +1,6 @@
 {
-  "last_node_id": 9,
-  "last_link_id": 15,
+  "last_node_id": 10,
+  "last_link_id": 19,
   "nodes": [
     {
       "id": 3,
@@ -20,7 +20,7 @@
         {
           "name": "source",
           "type": "*",
-          "link": 13,
+          "link": 18,
           "dir": 3
         }
       ],
@@ -49,7 +49,7 @@
         {
           "name": "source",
           "type": "*",
-          "link": 14,
+          "link": 17,
           "dir": 3
         }
       ],
@@ -78,7 +78,7 @@
         {
           "name": "source",
           "type": "*",
-          "link": 15,
+          "link": 16,
           "dir": 3
         }
       ],
@@ -107,7 +107,7 @@
         {
           "name": "images",
           "type": "IMAGE",
-          "link": 12
+          "link": 19
         }
       ],
       "properties": {
@@ -115,15 +115,15 @@
       }
     },
     {
-      "id": 8,
+      "id": 10,
       "type": "Eden_LoRa_trainer",
       "pos": [
-        454,
-        103
+        482,
+        102
       ],
       "size": {
         "0": 315,
-        "1": 430
+        "1": 454
       },
       "flags": {},
       "order": 0,
@@ -133,7 +133,7 @@
           "name": "sample_images",
           "type": "IMAGE",
           "links": [
-            12
+            19
           ],
           "shape": 3,
           "slot_index": 0
@@ -142,7 +142,7 @@
           "name": "lora_path",
           "type": "STRING",
           "links": [
-            13
+            18
           ],
           "shape": 3,
           "slot_index": 1
@@ -151,7 +151,7 @@
           "name": "embedding_path",
           "type": "STRING",
           "links": [
-            14
+            17
           ],
           "shape": 3,
           "slot_index": 2
@@ -160,7 +160,7 @@
           "name": "final_msg",
           "type": "STRING",
           "links": [
-            15
+            16
           ],
           "shape": 3,
           "slot_index": 3
@@ -172,13 +172,14 @@
       "widgets_values": [
         "https://storage.googleapis.com/public-assets-xander/A_workbox/lora_training_sets/clipx_tiny.zip",
         "zavychromaxl_v90.safetensors",
-        "clipx_style_test",
+        "Eden_LoRa",
         "style",
         4,
         360,
         0.001,
         0.001,
         16,
+        false,
         3,
         true,
         180,
@@ -190,12 +191,12 @@
       "id": 9,
       "type": "Note",
       "pos": [
-        465,
-        -245
+        500,
+        -204
       ],
       "size": [
-        833.2066162535789,
-        284.2696383134398
+        517.9482621039891,
+        235.97544808730578
       ],
       "flags": {},
       "order": 1,
@@ -204,7 +205,7 @@
         "text": ""
       },
       "widgets_values": [
-        "This LoRa trainer was made by the team behind eden.art, led by https://x.com/xsteenbrugge\nThis trainer works for both SD15 and SDXL models, but is primarily tuned for SDXL models.\n\nThis trainer optionally trains a textual inversion token into the LoRa, this is highly recommended when using SDXL models, but also means you have to load that token embedding when doing inference!\n\nThe default settings work great for SDXL, SD15 usually need more training steps (eg 1000) and usually benefits from disabling ti_training.\n\n\n\n\n\n"
+        "This LoRa trainer was made by the team behind eden.art, led by https://x.com/xsteenbrugge\nThis trainer works for both SD15 and SDXL models, but is primarily tuned for SDXL models.\n\nThis trainer optionally trains a textual inversion token into the LoRa, this is highly recommended when using SDXL models, but also means you have to load that token embedding when doing inference!\n\nThe default settings work great for SDXL, SD15 usually need more training steps (eg 1000) and usually benefits from disabling ti_training.\n\nIf you disbale_ti (not recommended) to obtain a normal LoRa that does not train a token embedding, you typically need to train for longer and increase the unet_lr like 2x-3x."
       ],
       "color": "#432",
       "bgcolor": "#653"
@@ -212,46 +213,46 @@
   ],
   "links": [
     [
-      12,
-      8,
-      0,
-      2,
-      0,
-      "IMAGE"
-    ],
-    [
-      13,
-      8,
-      1,
+      16,
+      10,
       3,
+      5,
       0,
       "*"
     ],
     [
-      14,
-      8,
+      17,
+      10,
       2,
       4,
       0,
       "*"
     ],
     [
-      15,
-      8,
+      18,
+      10,
+      1,
       3,
-      5,
       0,
       "*"
+    ],
+    [
+      19,
+      10,
+      0,
+      2,
+      0,
+      "IMAGE"
     ]
   ],
   "groups": [],
   "config": {},
   "extra": {
     "ds": {
-      "scale": 1.1,
+      "scale": 0.9090909090909091,
       "offset": [
-        -132.40529807176063,
-        133.74669805019636
+        -266.8482621039891,
+        347.82455191269423
       ]
     }
   },

trainer/config.py

@@ -26,7 +26,6 @@ model_paths = ModelPaths()
 # Default 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://huggingface.co/RunDiffusion/Juggernaut-XL-v6/resolve/main/juggernautXL_version6Rundiffusion.safetensors"
-#SDXL_URL = "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors"
 SD15_URL = "https://huggingface.co/KamCastle/jugg/resolve/main/juggernaut_reborn.safetensors"
 
 pretrained_models = {