Spaces:

rkihacker
/

R2OAI

Paused

rkihacker commited on Oct 21

Commit

13ff1c0

verified ·

1 Parent(s): 0f99721

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -56,8 +56,8 @@ class ChatCompletionChunk(BaseModel):
 # --- Supported Models ---
 SUPPORTED_MODELS = {
     "llama3-8b-instruct": "meta/meta-llama-3-8b-instruct",
-    "claude-3-haiku-20240307": "anthropic/claude-3-haiku-20240307", # Example of another common model
-    "claude-3-sonnet-20240229": "anthropic/claude-3-sonnet-20240229",
     "llava-13b": "yorickvp/llava-13b:e272157381e2a3bf12df3a8edd1f38d1dbd736bbb7437277c8b34175f8fce358"
 }

 # --- Supported Models ---
 SUPPORTED_MODELS = {
     "llama3-8b-instruct": "meta/meta-llama-3-8b-instruct",
+    "claude-4.5-haiku": "anthropic/claude-4.5-haiku",
+    "claude-4.5-sonnet": "anthropic/claude-4.5-sonnet",
     "llava-13b": "yorickvp/llava-13b:e272157381e2a3bf12df3a8edd1f38d1dbd736bbb7437277c8b34175f8fce358"
 }