diff --git a/mindspeed_llm/tasks/megatron_adaptor.py b/mindspeed_llm/tasks/megatron_adaptor.py index 8265c62cad6a5a5504d327df954fc3f4f2ef7893..00ba0651ce721d18f67c99ea83c11f32d0900bff 100644 --- a/mindspeed_llm/tasks/megatron_adaptor.py +++ b/mindspeed_llm/tasks/megatron_adaptor.py @@ -21,7 +21,7 @@ import torch import tensordict from torch_npu.contrib import transfer_to_npu from mindspeed_llm.features_manager import FEATURES_LIST - +#test def dummy_jit(fn): def wrapper(*args, **kwargs):