File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -1125,6 +1125,7 @@ def test_serialization_shared_tensors(self):
11251125 @require_non_cpu
11261126 @pytest .mark .single_gpu_tests
11271127 @require_bitsandbytes
1128+ @pytest .mark .skipif (not torch .cuda .is_available (), reason = "XPU have numerial errors" )
11281129 def test_4bit_dora_inference (self ):
11291130 # check for same result with and without DoRA when initializing with init_lora_weights=False
11301131 bnb_config = BitsAndBytesConfig (
@@ -1164,6 +1165,7 @@ def test_4bit_dora_inference(self):
11641165 @require_non_cpu
11651166 @pytest .mark .single_gpu_tests
11661167 @require_bitsandbytes
1168+ @pytest .mark .skipif (not torch .cuda .is_available (), reason = "XPU have numerial errors" )
11671169 def test_8bit_dora_inference (self ):
11681170 # check for same result with and without DoRA when initializing with init_lora_weights=False
11691171 model = AutoModelForCausalLM .from_pretrained (
You can’t perform that action at this time.
0 commit comments