From 8747c74339cf1e7f1d45f4aa1dcc090e9eba94a3 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 23 Mar 2023 22:19:01 -0300
Subject: [PATCH] Another missing import

---
 modules/callbacks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/callbacks.py b/modules/callbacks.py
index 40811408..2ae9d908 100644
--- a/modules/callbacks.py
+++ b/modules/callbacks.py
@@ -5,6 +5,8 @@ from threading import Thread
 import torch
 import transformers
 
+import modules.shared as shared
+
 
 # Copied from https://github.com/PygmalionAI/gradio-ui/
 class _SentinelTokenStoppingCriteria(transformers.StoppingCriteria):