Skip to content

Commit f7450e2

Browse files
committed
[bugfix] fix modelscope patch_hub (#6612)
1 parent 02a38ac commit f7450e2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

swift/megatron/init.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -669,6 +669,7 @@ def _apply_rotary_pos_emb_thd(
669669

670670

671671
def _patch_megatron():
672+
os.environ.pop('VLLM_USE_MODELSCOPE', None)
672673
logging_level = logging.root.level
673674
_patch_flash_attn()
674675
_patch_transformer_engine()

0 commit comments

Comments
 (0)