Spaces:

yabramuvdi
/

llms-demo

Running on Zero

yabramuvdi commited on Feb 21

Commit

c9abde4

verified ·

1 Parent(s): 93cbacc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,8 @@ AVAILABLE_MODELS = {
     "bloomz-560m": "bigscience/bloomz-560m",
     "gpt2-medium": "gpt2-medium",
     "opt-350m": "facebook/opt-350m",
-    "pythia-160m": "EleutherAI/pythia-160m"
 }
 # Initialize model and tokenizer

     "bloomz-560m": "bigscience/bloomz-560m",
     "gpt2-medium": "gpt2-medium",
     "opt-350m": "facebook/opt-350m",
+    "pythia-160m": "EleutherAI/pythia-160m",
+    "mistral-7b": "mistralai/Mistral-7B-Instruct"
 }
 # Initialize model and tokenizer