Fix import of FOCUS

konstantinjdobler · Nov 6, 2023 · b5ffdcf · b5ffdcf
1 parent 0b45d2a
commit b5ffdcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/focus_example.py b/focus_example.py
@@ -5,7 +5,7 @@
     PreTrainedTokenizer,
 )
 
-from focus import FOCUS
+from deepfocus import FOCUS
 
 source_tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
 source_model: PreTrainedModel = AutoModelForMaskedLM.from_pretrained("xlm-roberta-base")