diff --git a/modules/callbacks.py b/modules/callbacks.py index 2c04cc53..0f918f3d 100644 --- a/modules/callbacks.py +++ b/modules/callbacks.py @@ -1,11 +1,9 @@ -import gc import traceback from queue import Queue from threading import Thread import torch import transformers -from transformers import is_torch_npu_available, is_torch_xpu_available import modules.shared as shared