Skip to content

Commit 9360c0f

Browse files
committed
removed unused line
1 parent 4921c2c commit 9360c0f

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/torch/ptq/test_fq_lora.py

-3
Original file line numberDiff line numberDiff line change
@@ -99,9 +99,6 @@ def test_checkpoint_loading(tmp_path):
9999
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
100100
tokenizer = AutoTokenizer.from_pretrained(model_id)
101101
example_input = tokenizer("dummy", return_tensors="pt").to(device)
102-
ref_output = tokenizer.decode(
103-
model.generate(**example_input, do_sample=False, max_new_tokens=20)[0], skip_special_tokens=True
104-
)
105102
except_lm_head_and_5th_vproj = (
106103
r"^(?!.*(GPTNeoXLayer\[2\]/GPTNeoXSdpaAttention\[attention\]/Linear\[query_key_value\]/l|embed_out).*$).*$"
107104
)

0 commit comments

Comments
 (0)