ci

rreece · Aug 23, 2023 · 0f1b2c6 · 0f1b2c6
1 parent 8de6b01
commit 0f1b2c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/hf_wrappers/models/gpt2.py b/python/hf_wrappers/models/gpt2.py
@@ -17,7 +17,7 @@ class GPT2Handler(BaseHandler):
 
     def __init__(self, model_name=None):
         if model_name is None:
-            model_name = "gpt2-large"
+            model_name = "gpt2-medium"
         super().__init__(model_name=model_name)
 
     def setup_model(self, model_name):