Updated the version (#65)

* changed the version
* Update trainer.py

flagai/trainer.py

@@ -309,15 +309,9 @@ def get_dataloader(self, dataset, collate_fn, shuffle=False):
                                                shuffle=shuffle)
         else:
             if self.env_type == 'deepspeed+mpu':
-                # num_replicas = self.world_size // mpu.get_model_parallel_world_size(
-                # )
-                # rank = self.rank // mpu.get_model_parallel_world_size()
-                # rank = mpu.get_model_parallel_rank()
                 rank = mpu.get_model_parallel_src_rank()
-
                 sampler = torch.utils.data.distributed.DistributedSampler(
                     dataset,
-                    # num_replicas=num_replicas,
                     rank=rank,
                     shuffle=shuffle)
             else:

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="flagai",
-    version="v1.1.1",
+    version="v1.1.2",
     description="FlagAI aims to help researchers and developers to freely train and test large-scale models for NLP tasks.",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",