Update instruct_qa/generation/generator.py

McGill-NLP · May 29, 2024 · edd5983 · edd5983
1 parent 009df2d
commit edd5983
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/instruct_qa/generation/generator.py b/instruct_qa/generation/generator.py
@@ -74,7 +74,7 @@ def __init__(self, *args, **kwargs):
         }
 
         if completion_type is not None:
-            self.model_map[model_name] = completion_type
+            self.model_map[self.model_name] = completion_type
 
         assert (
             self.model_name in self.model_map