Update modeling_quiet.py

modeling_quiet.py

@@ -33,6 +33,7 @@ from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from transformers.generation.utils import GenerationMixin
 from transformers.generation.stopping_criteria import StoppingCriteriaList, validate_stopping_criteria
 from transformers import TextStreamer, AutoTokenizer
+import transformers
 
 from transformers.activations import ACT2FN
 from transformers.cache_utils import Cache, DynamicCache