Skip to content

Commit 4b99b24

Browse files
committed
Fix device variable
Signed-off-by: Andrea Fasoli <[email protected]>
1 parent 38b9093 commit 4b99b24

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fms_mo/quant/ptq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2420,7 +2420,7 @@ def get_act_scales(
24202420

24212421
for data_mb, _ in zip(pbar, range(n_samples)):
24222422
qcfg["sample_id"] += 1
2423-
data_mb = move_to(data_mb, dev)
2423+
data_mb = move_to(data_mb, device)
24242424
if (
24252425
qcfg["nbits_bmm1"] < 32
24262426
or qcfg["nbits_bmm2"] < 32

0 commit comments

Comments
 (0)