Fix minor naming bugs (#382)

Author: zhangbilang

llmc/models/internvl2.py

@@ -435,7 +435,7 @@ def __init__(
         elif accelerator.num_processes == 1 and device_map == 'auto':
             logger.info(f'Using {accelerator.num_processes} devices with tensor parallelism')
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         else:
             logger.info(f'Using single device: {self._device}')
             self.model.to(self._device)

llmc/models/llava.py

@@ -296,7 +296,7 @@ def __init__(
         elif accelerator.num_processes == 1 and device_map == 'auto':
             logger.info(f'Using {accelerator.num_processes} devices with tensor parallelism')
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         else:
             logger.info(f'Using single device: {self._device}')
             self.model.to(self._device)

llmc/models/llava_hf.py

@@ -254,10 +254,10 @@ def __init__(
                 f'Using {accelerator.num_processes} devices with pipeline parallelism'
             )
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         else:
             logger.info(f'Using single device: {self._device}')
             self.model.to(self._device)
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         self.accelerator = accelerator

llmc/models/minicpmv.py

@@ -174,7 +174,7 @@ def __init__(
         else:
             self.model.to(self._device)
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
 
     @property
     def config(self):

llmc/models/qwen2vl.py

@@ -275,7 +275,7 @@ def __init__(
                 self._world_size = self.accelerator.num_processes
             else:
                 self._rank = 0
-                self._word_size = 1
+                self._world_size = 1
 except Exception:
     logger.warning(
         'Can not import lmms_eval. '

llmc/models/videollava.py

@@ -157,7 +157,7 @@ def __init__(
                 f'Using {accelerator.num_processes} devices with tensor parallelism'
             )
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         else:
             logger.info(f'Using single device: {self._device}')
             self.model.to(self._device)

llmc/models/vila.py

@@ -302,7 +302,7 @@ def __init__(
                 f'Using {accelerator.num_processes} devices with tensor parallelism'
             )
             self._rank = 0
-            self._word_size = 1
+            self._world_size = 1
         else:
             logger.info(f'Using single device: {self._device}')
             self._model.to(self._device)