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

2
modules/GPTQ_loader.py
View file

@ -5,7 +5,7 @@ from pathlib import Path
import accelerate
import torch
import transformers
from accelerate import is_xpu_available
from accelerate.utils import is_xpu_available
from gptq_for_llama import llama_inference_offload
from gptq_for_llama.modelutils import find_layers
from gptq_for_llama.quant import make_quant