diff --git a/community_resources/max_tokens.json b/community_resources/max_tokens.json
index e61239f2c..543394dfe 100644
--- a/community_resources/max_tokens.json
+++ b/community_resources/max_tokens.json
@@ -12,5 +12,5 @@
     "text-bison-001": {"max_tokens": 8192, "input_cost_per_token": 0.000004, "output_cost_per_token": 0.000004},
     "chat-bison-001": {"max_tokens": 4096, "input_cost_per_token": 0.000002, "output_cost_per_token": 0.000002},
     "command-nightly": {"max_tokens": 4096, "input_cost_per_token": 0.000015, "output_cost_per_token": 0.000015},
-    "replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1": {"max_tokens": 4096, "input_cost_per_token": 0.00000608, "output_cost_per_token": 0.00000608},
+    "replicate/llama-2-70b-chat:2c1608e18606fad2812020dc541930f2d0495ce32eee50074220b87300bc16e1": {"max_tokens": 4096, "input_cost_per_token": 0.00000608, "output_cost_per_token": 0.00000608}
 }}
\ No newline at end of file
diff --git a/litellm/__init__.py b/litellm/__init__.py
index 3efd14343..01bfc49b5 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -47,6 +47,10 @@ open_ai_text_completion_models = [
 
 cohere_models = [
     'command-nightly',
+    "command", 
+    "command-light", 
+    "command-medium-beta", 
+    "command-xlarge-beta"
 ]
 
 anthropic_models = [