This commit is contained in:
oobabooga 2025-01-09 13:18:23 -08:00
parent 0e94d7075e
commit f3c0f964a2

View File

@ -1,11 +1,9 @@
import gc
import traceback import traceback
from queue import Queue from queue import Queue
from threading import Thread from threading import Thread
import torch import torch
import transformers import transformers
from transformers import is_torch_npu_available, is_torch_xpu_available
import modules.shared as shared import modules.shared as shared