Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelbenayoun committed Feb 15, 2024
1 parent 11b2797 commit 2e0917c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/neuron/utils/training_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -314,7 +314,7 @@ def set_neuron_cc_optlevel_for_model(model: "PreTrainedModel", optlevel: str = "
if match_:
neuron_cc_flags = neuron_cc_flags[: match_.start(0)] + f"{optlevel}" + neuron_cc_flags[match_.end(0) + 1 :]
else:
neuron_cc_flags += f"{optlevel} "
neuron_cc_flags += f" {optlevel} "
os.environ["NEURON_CC_FLAGS"] = neuron_cc_flags


Expand Down

0 comments on commit 2e0917c

Please sign in to comment.