add llama option in monash

2132660698 · Oct 11, 2023 · a27bf7a · a27bf7a
1 parent c77340a
commit a27bf7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/experiments/run_monash.py b/experiments/run_monash.py
@@ -22,11 +22,13 @@
 
 model_hypers = {
     'text-davinci-003': {'model': 'text-davinci-003', **gpt3_hypers},
+    'llama-70b': {'model': 'llama-70b', **llama_hypers},
 }
 
 # Specify the function to get predictions for each model
 model_predict_fns = {
     'text-davinci-003': get_llmtime_predictions_data,
+    'llama-70b': get_llmtime_predictions_data,
 }
 
 def is_gpt(model):