Spaces:
Sleeping
Sleeping
File size: 1,829 Bytes
32c5da4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | """Check which providers are available"""
import os
import sys
# Set HuggingFace cache
os.environ['HF_HOME'] = 'd:/VSC Codes/Bild/.cache/hf'
os.environ['TRANSFORMERS_CACHE'] = 'd:/VSC Codes/Bild/.cache/hf'
print("Checking providers...")
print()
# Check LocalAI
try:
from backend.app.local_ai.engine import LocalAIEngine
engine = LocalAIEngine()
available = engine.is_available()
print(f"LocalAI Engine:")
print(f" Available: {available}")
print(f" Model ID: {engine.model_id}")
if available:
try:
print(f" Attempting to load model...")
pipe = engine._ensure()
print(f" ✓ Model loaded successfully!")
print(f" Device: {pipe.device}")
except Exception as e:
print(f" ✗ Failed to load: {e}")
except Exception as e:
print(f"LocalAI: ✗ Error: {e}")
print()
# Check Diffusion Provider
try:
from backend.app.providers.diffusion_provider import DiffusionProvider
provider = DiffusionProvider()
available = provider.is_available()
print(f"Diffusion Provider:")
print(f" Available: {available}")
if available:
try:
print(f" Testing generation capability...")
# Don't actually generate, just check if provider thinks it can
print(f" ✓ Provider ready")
except Exception as e:
print(f" ✗ Error: {e}")
except Exception as e:
print(f"Diffusion Provider: ✗ Error: {e}")
print()
# List all registered providers
try:
from backend.app.providers.registry import list_available_providers
providers = list_available_providers()
print(f"Registered available providers: {', '.join(providers) if providers else 'NONE (will use dummy!)'}")
except Exception as e:
print(f"Registry check failed: {e}")
|