From cb587e0336ee2f49da2fb808d5d8674d497a02e8 Mon Sep 17 00:00:00 2001 From: alex-lew Date: Tue, 23 Jul 2024 20:30:55 +0000 Subject: [PATCH] Add import for defaultdict --- hfppl/llms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hfppl/llms.py b/hfppl/llms.py index 6c10012..66abca4 100644 --- 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: