Skip to content

Commit 6d4ad3f

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 3a77dd2 commit 6d4ad3f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,9 @@
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, no-name-in-module
23+
from trl import ( # pylint: disable=import-error, no-name-in-module
24+
DataCollatorForCompletionOnlyLM,
25+
)
2426
import torch
2527

2628

0 commit comments

Comments
 (0)