Initial Commit

2024-06-02 14:51:22 +08:00
commit af8096b4b4
11 changed files with 283 additions and 0 deletions
@@ -0,0 +1,39 @@
+import os
+import utils
+
+# Define the base model path
+base_model_path = "C:\\Users\\Admin\\Desktop\\ComfyUI_windows_portable\\ComfyUI\\models\\diffusers\\raemumix_v90"
+
+# Test loading the main model
+try:
+    model = utils.load_model_from_path(os.path.join(base_model_path, "model_index.json"))
+    print("Main model loaded successfully.")
+except Exception as e:
+    print(f"Error loading main model: {e}")
+
+# Test loading the VAE model
+try:
+    vae = utils.load_vae_from_path(base_model_path)
+    print("VAE model loaded successfully.")
+except Exception as e:
+    print(f"Error loading VAE model: {e}")
+
+# Test loading the CLIP model
+try:
+    clip = utils.load_text_encoder_from_path(base_model_path, encoder_id=None)
+    print("CLIP model loaded successfully.")
+except Exception as e:
+    print(f"Error loading CLIP model: {e}")
+
+# Test loading the text encoders for SDXL
+try:
+    text_encoder_1 = utils.load_text_encoder_from_path(base_model_path, 1)
+    print("Text encoder 1 loaded successfully.")
+except Exception as e:
+    print(f"Error loading text encoder 1: {e}")
+
+try:
+    text_encoder_2 = utils.load_text_encoder_from_path(base_model_path, 2)
+    print("Text encoder 2 loaded successfully.")
+except Exception as e:
+    print(f"Error loading text encoder 2: {e}")