chroma support tweaks

Author: mcmonkey4eva

docs/Model Support.md

@@ -285,7 +285,9 @@ Parameters and usage is the same as any other normal model.
 ### Chroma
 
 - Chroma is a derivative of Flux, and is supported in SwarmUI
-    - Downloads here <https://huggingface.co/lodestones/Chroma/tree/main>
+    - FP8 Scaled versions here: <https://huggingface.co/Clybius/Chroma-fp8-scaled/tree/main>
+    - Or GGUF versions here: <https://huggingface.co/silveroxides/Chroma-GGUF>
+    - Or original BF16 here (not recommended): <https://huggingface.co/lodestones/Chroma/tree/main>
     - Uses standard CFG, not distilled to 1 like other Flux models
     - Official reference workflow uses Scheduler=`Align Your Steps` with Steps=`26` and CFG Scale=`4`
     - Probably works better with longer prompts

src/BuiltinExtensions/ComfyUIBackend/WorkflowGenerator.cs

@@ -918,6 +918,10 @@ void doVaeLoader(string defaultVal, string compatClass, string knownName)
                     {
                         dtype = "default";
                     }
+                    else if (model.Metadata?.SpecialFormat == "fp8_scaled")
+                    {
+                        dtype = "default";
+                    }
                     else
                     {
                         dtype = "fp8_e4m3fn";

src/Text2Image/T2IModelHandler.cs

@@ -509,6 +509,11 @@ static string[] procWordsFrom(JToken tok)
                     specialFormat = "bnb_nf4";
                     break;
                 }
+                if (key.EndsWith(".scale_weight"))
+                {
+                    specialFormat = "fp8_scaled";
+                    break;
+                }
             }
             if (model.Name.EndsWith(".gguf"))
             {