Fix model name

language_model.py

@@ -51,9 +51,9 @@ class TransformersAVG:
                  ) -> None:
         
         self.is_ag = 'ag' in model.split('-') if is_ag is None else is_ag
-        self.is_ve = 'ae' in model.split('-') if is_ve is None else is_ve
-        self.is_av = 'av' in model.split('-') if is_av is None else is_av
-        self.is_avg = 'avg' in model.split('-') if is_avg is None else is_avg
+        self.is_ve = 've' in model.split('-') if is_ve is None else is_ve
+        self.is_av = 'mlt' in model.split('-') if is_av is None else is_av
+        self.is_avg = 'end2end' in model.split('-') if is_avg is None else is_avg
 
         self.model_name = model
         self.max_input_length = max_input_length