From f421586a6990aa18b2ffe501a96c2e1a0cd47cae Mon Sep 17 00:00:00 2001
From: Marcin Ostrowski <marcinostrowski97@outlook.com>
Date: Tue, 7 Jan 2025 20:05:25 +0100
Subject: [PATCH] Add missing DEFAULT_TRAIN_BSIZE field

---
 torchbenchmark/models/cm3leon_generate/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torchbenchmark/models/cm3leon_generate/__init__.py b/torchbenchmark/models/cm3leon_generate/__init__.py
index 4bd3ee8ed2..d36d16c523 100644
--- a/torchbenchmark/models/cm3leon_generate/__init__.py
+++ b/torchbenchmark/models/cm3leon_generate/__init__.py
@@ -7,6 +7,7 @@
 
 class Model(BenchmarkModel):
     task = NLP.LANGUAGE_MODELING
+    DEFAULT_TRAIN_BSIZE = 1
     DEFAULT_EVAL_BSIZE = 1
 
     def __init__(self, test, device, batch_size=None, extra_args=[]):