Skip to content

Commit 3a77dd2

Browse files
committed
fix: update global variable syntax since its not support from torch 3.2.0
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
1 parent 7ad5a50 commit 3a77dd2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

plugins/attention-and-distributed-packing/src/fms_acceleration_aadp/framework_plugin_padding_free.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from fms_acceleration import AccelerationPlugin
2121
from peft import LoraConfig
2222
from transformers import DataCollatorForSeq2Seq, TrainingArguments
23-
from trl import DataCollatorForCompletionOnlyLM # pylint: disable=import-error
23+
from trl import DataCollatorForCompletionOnlyLM # pylint: disable=import-error, no-name-in-module
2424
import torch
2525

2626

0 commit comments

Comments
 (0)