Code

[shardformer]gather llama logits#5398

Code

Merged

flybird11111 merged 2 commits intohpcaitech:mainhpcaitech/ColossalAI:mainfrom flybird11111:llama-gatherflybird11111/ColossalAI:llama-gatherCopy head branch name to clipboardFeb 27, 2024

Commits

Commits on Feb 26, 2024

• gather llama logits

  flybird11111

  committed
  5173ded

  Copy full SHA for 5173ded

Commits on Feb 27, 2024

• fix

  flybird11111

  authored
  14f87ea

  Copy full SHA for 14f87ea