Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

refactor pytorch engine(ascend) #2440

Merged
merged 11 commits into from
Sep 11, 2024

Conversation

yao-fengchen
Copy link
Collaborator

  1. Refactor pytorch engine(ascend) based on Refactor pytorch engine #2104
  2. The models we currently support on ascend include:
    • llama-2-7b-hf
    • llama-2-70b-chat-hf
    • Llama-3-8B-Instruct
    • Llama-3.1-8B-Instruct
    • internlm2-chat-7b
    • internlm2-chat-20b
    • internlm2_5-7b-chat
    • internlm2_5-20b-chat
    • InternVL-Chat-V1-5
    • InternVL2-2B
    • Mixtral-8x7B-Instruct-v0.1
    • Qwen2-7B-Instruct
    • Qwen2-57B-A14B-Instruct

@yao-fengchen yao-fengchen force-pushed the yfc/ascend-torch-layers branch from e805d2f to 1e2665c Compare September 10, 2024 10:38
@lvhan028 lvhan028 requested a review from grimoire September 11, 2024 04:31
@lvhan028 lvhan028 added the enhancement New feature or request label Sep 11, 2024
@lvhan028 lvhan028 merged commit e681077 into InternLM:main Sep 11, 2024
5 checks passed
@yao-fengchen yao-fengchen deleted the yfc/ascend-torch-layers branch September 12, 2024 02:02
@yao-fengchen yao-fengchen restored the yfc/ascend-torch-layers branch September 12, 2024 02:03
@yao-fengchen yao-fengchen deleted the yfc/ascend-torch-layers branch November 19, 2024 05:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants