amankhandelia
/

panini

Model card Files Files and versions

hassiahk commited on Jul 17, 2021

Commit

d5cc6ba

·

1 Parent(s): c4e4546

Convert latest model to PyTorch

Files changed (2) hide show

flax_to_torch.py +1 -1
pytorch_model.bin +2 -2

flax_to_torch.py CHANGED Viewed

@@ -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"
     )
     parser.add_argument("--torch_model_path", type=str, default="./", help="PyTorch model path")

 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description="Flax to Pytorch model coversion")
     parser.add_argument(
+        "--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 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:6a12ed7389606b65846e3445153a010efbada4755ea9fbb1f558b347efcdfc57
+size 498872747