| |
| """ |
| Verification script to show current model configuration |
| """ |
| import os |
|
|
| def show_model_config(): |
| """Show what model will be used""" |
| print("🔍 Model Configuration Analysis") |
| print("=" * 50) |
| |
| |
| ai_model_env = os.environ.get("AI_MODEL", "google/gemma-3n-E4B-it") |
| print(f"📝 Environment variable AI_MODEL: {ai_model_env}") |
| |
| |
| if "GGUF" in ai_model_env: |
| current_model = "google/gemma-3n-E4B-it" |
| print(f"🔄 OVERRIDE: GGUF model detected, using: {current_model}") |
| print(f" Original: {ai_model_env}") |
| print(f" Fixed to: {current_model}") |
| else: |
| current_model = ai_model_env |
| print(f"✅ Using: {current_model}") |
| |
| print(f"\n🎯 Final model that will be loaded: {current_model}") |
| |
| |
| is_gemma_3n = "gemma-3n" in current_model.lower() |
| print(f"🤖 Is Gemma 3n model: {is_gemma_3n}") |
| |
| if is_gemma_3n: |
| print("📚 Will use: AutoProcessor + Gemma3nForConditionalGeneration") |
| else: |
| print("📚 Will use: AutoTokenizer + AutoModelForCausalLM") |
| |
| return current_model |
|
|
| if __name__ == "__main__": |
| show_model_config() |
|
|
