Update convert_roformer_original_tf_checkpoint_to_pytorch.py

JunnYu · web-flow · commit c214e0a36a01 · 2021-05-18T09:49:15.000+08:00
diff --git a/src/roformer/convert_roformer_original_tf_checkpoint_to_pytorch.py b/src/roformer/convert_roformer_original_tf_checkpoint_to_pytorch.py
@@ -19,7 +19,7 @@
 
 import torch
 
-from transformers import RoFormerConfig, RoFormerForMaskedLM, load_tf_weights_in_roformer
+from roformer import RoFormerConfig, RoFormerForMaskedLM, load_tf_weights_in_roformer
 from transformers.utils import logging
 
 
