Back out "fix bert example" (#728)

Summary: Pull Request resolved: #728 ATT Reviewed By: alexanderguzhva Differential Revision: D46186394 fbshipit-source-id: 01a41450ebe5a07e963da131c764731b3a6f0ce3
facebookincubator · May 25, 2023 · 02460f6 · 02460f6
1 parent 201937e
commit 02460f6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/examples/03_bert/benchmark_ait.py b/examples/03_bert/benchmark_ait.py
@@ -24,8 +24,8 @@
 from aitemplate.frontend import Tensor
 from aitemplate.testing import detect_target
 
-from modeling.bert import BertBaseEncodersOnly, BertBaseUncased
-from modeling.torch_model import BertBaseUncased as BertPt
+from .modeling.bert import BertBaseEncodersOnly, BertBaseUncased
+from .modeling.torch_model import BertBaseUncased as BertPt
 
 
 def mark_output(y: Tensor) -> None:

diff --git a/examples/03_bert/demo.py b/examples/03_bert/demo.py
@@ -16,11 +16,11 @@
 
 import torch
 
-from benchmark_ait import compile_module
-from modeling.torch_model import BertBaseUncased as BertPt
-
 from transformers import BertTokenizer
 
+from .benchmark_ait import compile_module
+from .modeling.torch_model import BertBaseUncased as BertPt
+
 
 def prepare_data(prompt: str, model_path: str):
     tokenizer = BertTokenizer.from_pretrained(model_path)

diff --git a/examples/03_bert/test_correctness.py b/examples/03_bert/test_correctness.py
@@ -18,7 +18,7 @@
 
 import torch
 
-from demo import run_model
+from .demo import run_model
 
 try:
     from libfb.py.asyncio.await_utils import await_sync