Spaces:

multimodalart
/

civitai-to-hf

Running

multimodalart HF staff commited on Aug 14

Commit

93009ec

•

1 Parent(s): cfe8a79

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def get_prompts_from_image(image_id):
 def extract_info(json_data):
     if json_data["type"] == "LORA":
         for model_version in json_data["modelVersions"]:
-            if model_version["baseModel"] in ["SDXL 1.0", "SDXL 0.9", "SD 1.5", "SD 1.4", "SD 2.1", "SD 2.0", "SD 2.0 768", "SD 2.1 768"]:
                 for file in model_version["files"]:
                     print(file)
                     if "primary" in file:

 def extract_info(json_data):
     if json_data["type"] == "LORA":
         for model_version in json_data["modelVersions"]:
+            if model_version["baseModel"] in ["SDXL 1.0", "SDXL 0.9", "SD 1.5", "SD 1.4", "SD 2.1", "SD 2.0", "SD 2.0 768", "SD 2.1 768", "Flux.1 D", "Flux.1 S"]:
                 for file in model_version["files"]:
                     print(file)
                     if "primary" in file: