58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Check available checkpoints in ComfyUI
|
|
"""
|
|
import requests
|
|
import json
|
|
|
|
COMFYUI_URL = "http://127.0.0.1:8000"
|
|
|
|
print("=" * 70)
|
|
print("🔍 ComfyUI - Available Checkpoints")
|
|
print("=" * 70)
|
|
|
|
# Get object info to see available checkpoints
|
|
try:
|
|
response = requests.get(f"{COMFYUI_URL}/object_info")
|
|
if response.status_code == 200:
|
|
data = response.json()
|
|
|
|
# Find CheckpointLoaderSimple
|
|
if "CheckpointLoaderSimple" in data:
|
|
checkpoint_info = data["CheckpointLoaderSimple"]
|
|
if "input" in checkpoint_info and "required" in checkpoint_info["input"]:
|
|
ckpt_list = checkpoint_info["input"]["required"].get("ckpt_name", [[]])[0]
|
|
|
|
print(f"\n✅ Available Checkpoints ({len(ckpt_list)}):")
|
|
for i, ckpt in enumerate(ckpt_list, 1):
|
|
print(f" {i}. {ckpt}")
|
|
|
|
if not ckpt_list:
|
|
print("\n❌ NO CHECKPOINTS FOUND!")
|
|
print("You need to download a model to:")
|
|
print("/Users/davidkotnik/Documents/ComfyUI/models/checkpoints/")
|
|
print("\nRecommended models:")
|
|
print("- SD 1.5: https://huggingface.co/runwayml/stable-diffusion-v1-5")
|
|
print("- SDXL: https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0")
|
|
else:
|
|
print("❌ Could not find checkpoint list structure")
|
|
else:
|
|
print("❌ CheckpointLoaderSimple not found in object_info")
|
|
|
|
else:
|
|
print(f"❌ Error: {response.status_code}")
|
|
|
|
except Exception as e:
|
|
print(f"❌ Exception: {e}")
|
|
|
|
# Also check for other info
|
|
print("\n" + "=" * 70)
|
|
print("📦 Checking /embeddings endpoint...")
|
|
try:
|
|
response = requests.get(f"{COMFYUI_URL}/embeddings")
|
|
if response.status_code == 200:
|
|
embeddings = response.json()
|
|
print(f"✅ Found {len(embeddings)} embeddings")
|
|
except Exception as e:
|
|
print(f"⚠️ {e}")
|