Fix is_ccl_available & is_xpu_available imports

modules/AutoGPTQ_loader.py

@@ -1,6 +1,6 @@
 from pathlib import Path
 
-from accelerate import is_xpu_available
+from accelerate.utils import is_xpu_available
 from auto_gptq import AutoGPTQForCausalLM, BaseQuantizeConfig
 
 import modules.shared as shared

modules/GPTQ_loader.py

@@ -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

modules/models.py

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

requirements.txt

@@ -1,4 +1,4 @@
-accelerate==0.23.*
+accelerate==0.24.*
 colorama
 datasets
 einops