TST Add missing .eval() calls to inference tests (#2408)

faaany · web-flow · commit 8edaae9460e4 · 2025-03-07T16:59:19.000+01:00
diff --git a/tests/test_common_gpu.py b/tests/test_common_gpu.py
@@ -1153,7 +1153,7 @@ def test_4bit_dora_inference(self):
         )
         torch.manual_seed(0)
         config_dora = LoraConfig(r=8, init_lora_weights=False, use_dora=True)
-        model = get_peft_model(model, config_dora)
+        model = get_peft_model(model, config_dora).eval()
 
         logits_dora = model(random_input).logits
 
@@ -1187,7 +1187,7 @@ def test_8bit_dora_inference(self):
         )
         torch.manual_seed(0)
         config_dora = LoraConfig(r=8, init_lora_weights=False, use_dora=True)
-        model = get_peft_model(model, config_dora)
+        model = get_peft_model(model, config_dora).eval()
 
         logits_dora = model(random_input).logits
 

Original file line number	Diff line number	Diff line change
`@@ -1153,7 +1153,7 @@ def test_4bit_dora_inference(self):`
`1153`	`1153`	`)`
`1154`	`1154`	`torch.manual_seed(0)`
`1155`	`1155`	`config_dora = LoraConfig(r=8, init_lora_weights=False, use_dora=True)`
`1156`		`- model = get_peft_model(model, config_dora)`
	`1156`	`+ model = get_peft_model(model, config_dora).eval()`
`1157`	`1157`
`1158`	`1158`	`logits_dora = model(random_input).logits`
`1159`	`1159`
`@@ -1187,7 +1187,7 @@ def test_8bit_dora_inference(self):`
`1187`	`1187`	`)`
`1188`	`1188`	`torch.manual_seed(0)`
`1189`	`1189`	`config_dora = LoraConfig(r=8, init_lora_weights=False, use_dora=True)`
`1190`		`- model = get_peft_model(model, config_dora)`
	`1190`	`+ model = get_peft_model(model, config_dora).eval()`
`1191`	`1191`
`1192`	`1192`	`logits_dora = model(random_input).logits`
`1193`	`1193`