Convert latest model to PyTorch

flax_to_torch.py

@@ -14,7 +14,7 @@ def convert_flax_model_to_torch(flax_model_path: str, torch_model_path: str = ".
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Flax to Pytorch model coversion")
     parser.add_argument(
-        "--flax_model_path", type=str, default="flax-community/roberta-pretraining-hindi", help="Flax model path"
+        "--flax_model_path", type=str, default="flax-community/roberta-hindi", help="Flax model path"
     )
     parser.add_argument("--torch_model_path", type=str, default="./", help="PyTorch model path")
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b875bdd3dd97d43498691f8fa783a8ded0881b387151102d6fe15b0818b596f
-size 498858859
+oid sha256:6a12ed7389606b65846e3445153a010efbada4755ea9fbb1f558b347efcdfc57
+size 498872747