test_localai.py

#!/usr/bin/env python3
"""Test if LocalAI can load tiny-sd model"""
import os
import sys

os.environ['HF_HOME'] = 'd:/VSC Codes/Bild/.cache/hf'
os.environ['TRANSFORMERS_CACHE'] = 'd:/VSC Codes/Bild/.cache/hf/transformers'
os.chdir('d:/VSC Codes/Bild/imageforge')

print("Testing LocalAI engine...")
print()

try:
    from backend.app.local_ai.engine import LocalAIEngine
    
    engine = LocalAIEngine()
    print(f"Model ID: {engine.model_id}")
    print(f"Available: {engine.is_available()}")
    print()
    
    if engine.is_available():
        print("Attempting to load model...")
        pipe = engine._ensure()
        print(f"✓ Model loaded successfully!")
        print(f"  Device: {pipe.device}")
        print(f"  Pipeline: {type(pipe).__name__}")
        sys.exit(0)
    else:
        print("✗ LocalAI not available (torch/diffusers not loaded)")
        sys.exit(1)
        
except Exception as e:
    print(f"✗ Error: {type(e).__name__}: {e}")
    import traceback
    traceback.print_exc()
    sys.exit(1)