40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
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}")
|