diff --git a/arch/tools/cli/config_generator.py b/arch/tools/cli/config_generator.py
index 3ebbb48f..0de077a8 100644
--- a/arch/tools/cli/config_generator.py
+++ b/arch/tools/cli/config_generator.py
@@ -19,6 +19,7 @@ SUPPORTED_PROVIDERS = [
     "together_ai",
     "azure_openai",
     "xai",
+    "ollama",
 ]
 
 
@@ -219,10 +220,12 @@ def validate_and_render_schema():
                 )
             provider = model_name_tokens[0]
 
-            # Validate azure_openai provider requires base_url
-            if provider == "azure_openai" and llm_provider.get("base_url") is None:
+            # Validate azure_openai and ollama provider requires base_url
+            if (
+                provider == "azure_openai" or provider == "ollama"
+            ) and llm_provider.get("base_url") is None:
                 raise Exception(
-                    f"Provider 'azure_openai' requires 'base_url' to be set for model {model_name}"
+                    f"Provider '{provider}' requires 'base_url' to be set for model {model_name}"
                 )
 
             model_id = "/".join(model_name_tokens[1:])