Add import for defaultdict

probcomp · Jul 23, 2024 · cb587e0 · cb587e0
1 parent f6efcd5
commit cb587e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hfppl/llms.py b/hfppl/llms.py
@@ -7,6 +7,7 @@
 from transformers import AutoModelForCausalLM
 from transformers import AutoTokenizer
 from transformers import BitsAndBytesConfig
+from collections import defaultdict
 
 
 class Masks: