Skip to content

Commit 9409a2f

Browse files
committed
Apply isort and black reformatting
Signed-off-by: artbataev <[email protected]>
1 parent 5ab58e1 commit 9409a2f

File tree

4 files changed

+4
-0
lines changed

4 files changed

+4
-0
lines changed

examples/llm/peft/automodel.py

+1
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ def main():
172172
jit_config = None
173173
if args.use_torch_jit:
174174
from nemo.automodel.compiler import TorchCompileConfig
175+
175176
jit_config = TorchCompileConfig(kwargs={'dynamic': True})
176177

177178
if args.use_te_optimizer:

examples/llm/sft/automodel.py

+1
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ def main():
171171
jit_config = None
172172
if args.use_torch_jit:
173173
from nemo.automodel.compiler import TorchCompileConfig
174+
174175
jit_config = TorchCompileConfig(kwargs={'dynamic': True})
175176

176177
if args.use_te_optimizer:

tests/collections/llm/hf/peft_hf.py

+1
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,7 @@ def main():
281281
jit_config = None
282282
if args.use_torch_jit:
283283
from nemo.automodel.compiler import TorchCompileConfig
284+
284285
jit_config = TorchCompileConfig(kwargs={'dynamic': True})
285286

286287
model = llm.HFAutoModelForCausalLM(model_name=args.model, compiler_config=jit_config)

tests/collections/llm/hf/sft.py

+1
Original file line numberDiff line numberDiff line change
@@ -324,6 +324,7 @@ def main():
324324
jit_config = None
325325
if args.use_torch_jit:
326326
from nemo.automodel.compiler import TorchCompileConfig
327+
327328
jit_config = TorchCompileConfig(kwargs={'dynamic': True})
328329

329330
if args.auto_resume:

0 commit comments

Comments
 (0)