Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save alexanderguzhva/1c48ba6dc74b03bf8538b0ce5f5611e9 to your computer and use it in GitHub Desktop.
Save alexanderguzhva/1c48ba6dc74b03bf8538b0ce5f5611e9 to your computer and use it in GitHub Desktop.
diff --git a/examples/03_bert/demo.py b/examples/03_bert/demo.py
--- a/examples/03_bert/demo.py
+++ b/examples/03_bert/demo.py
@@ -16,11 +16,11 @@
import torch
-from transformers import BertTokenizer
-
from benchmark_ait import compile_module
from modeling.torch_model import BertBaseUncased as BertPt
+from transformers import BertTokenizer
+
def prepare_data(prompt: str, model_path: str):
tokenizer = BertTokenizer.from_pretrained(model_path)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment