Skip to content

GitLab

Commit a45f4c2a authored by wxj's avatar wxj
Browse files

Update language_model.py

parent b2f70020
Pipeline #2226 passed with stage
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
megatron/legacy/model/language_model.py
View file @ a45f4c2a
...@@ -471,7 +471,7 @@ class TransformerLanguageModel(MegatronModule): ...@@ -471,7 +471,7 @@ class TransformerLanguageModel(MegatronModule):
else: else:
raise Exception('Stage must have at least either encoder or decoder') raise Exception('Stage must have at least either encoder or decoder')
@torch.compile(mode="max-autotune-no-cudagraphs") # @torch.compile(mode="max-autotune-no-cudagraphs")
def forward( def forward(
self, self,
enc_input_ids, enc_input_ids,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment