Update civitaiinfo_convert.py

civitaiinfo_convert.py

@@ -18,13 +18,21 @@ for file in glob.glob("**/*.civitai.info", recursive=True):
     print(jsonname)
     with open(file) as f:
         info = json.load(f)
-        
+    
+    sdversion = "Unknown"
+    if 'SD 1' in info['baseModel']:
+        sdversion = "SD1"
+    if 'SD 2' in info['baseModel']:
+        sdversion = "SD2"
+    if 'SDXL' in info['baseModel']:
+        sdversion = "SDXL"
 #    out["description"] = info.get("description", "")
 #    if not out["description"]:
     out["description"] = description
     out["activation text"] = ", ".join(info["trainedWords"])
     out["notes"] = f"https://civitai.com/models/{info['modelId']}?modelVersionId={info['id']}"
-    
+    out["sd version"] = sdversion 
+    out["preferred weight"] = 1
     print(jsonname)
     with open(jsonname, "w") as f:
         json.dump(out, f, indent=4)