mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix is_ccl_available & is_xpu_available imports

Browse source
This commit is contained in:
oobabooga 2023-10-26 20:26:25 -07:00
parent 778a010df8
commit 839a87bac8
4 changed files with 5 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

8
modules/models.py
View file

@ -7,12 +7,8 @@ from pathlib import Path
import torch
import transformers
from accelerate import (
infer_auto_device_map,
init_empty_weights,
is_ccl_available,
is_xpu_available
)
from accelerate import infer_auto_device_map, init_empty_weights
from accelerate.utils import is_ccl_available, is_xpu_available
from transformers import (
AutoConfig,
AutoModel,