Another missing import

2023-03-23 22:19:01 -03:00
parent 7078d168c3
commit 8747c74339
1 changed files with 2 additions and 0 deletions
--- 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):