Skip to content

Commit 63bc9eb

Browse files
committed
Fixed recent comment
Signed-off-by: Meet Patel <[email protected]>
1 parent 66b41d4 commit 63bc9eb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

QEfficient/cloud/finetune.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def main(**kwargs):
110110
# Enable gradient checkpointing
111111
if train_config.gradient_checkpointing:
112112
# Note: below attribute and method is only available in HuggingFace Transformer models.
113-
if model.supports_gradient_checkpointing:
113+
if hasattr(model, "supports_gradient_checkpointing") and model.supports_gradient_checkpointing:
114114
model.gradient_checkpointing_enable(gradient_checkpointing_kwargs={"preserve_rng_state": False})
115115
else:
116116
raise RuntimeError("Given model doesn't support gradient checkpointing. Please disable it and run it.")

0 commit comments

Comments
 (0)