Skip to content

[example] Modify palm example with the new booster API#3913

Merged
FrankLeeeee merged 6 commits intohpcaitech:mainfrom
MaruyamaAya:palm2
Jun 7, 2023
Merged

[example] Modify palm example with the new booster API#3913
FrankLeeeee merged 6 commits intohpcaitech:mainfrom
MaruyamaAya:palm2

Conversation

@MaruyamaAya
Copy link
Copy Markdown
Contributor

@MaruyamaAya MaruyamaAya commented Jun 6, 2023

📌 Checklist before creating the PR

  • I have created an issue for this PR for traceability
  • The title follows the standard format: [doc/gemini/tensor/...]: A concise description
  • I have added relevant tags if possible for us to better distinguish different PRs

🚨 Issue number

Link this PR to your issue with words like fixed to automatically close the linked issue upon merge

e.g. fixed #1234, closed #1234, resolved #1234

📝 What does this PR do?

Modify palm example with the new booster API

Summarize your work here.
if you have any plots/diagrams/screenshots/tables, please attach them here.

💥 Checklist before requesting a review

  • I have linked my PR to an issue (instruction)
  • My issue clearly describes the problem/feature/proposal, with diagrams/charts/table/code if possible
  • I have performed a self-review of my code
  • I have added thorough tests.
  • I have added docstrings for all the functions/methods I implemented

⭐️ Do you enjoy contributing to Colossal-AI?

  • 🌝 Yes, I do.
  • 🌚 No, I don't.

Tell us more if you don't enjoy contributing to Colossal-AI.

@MaruyamaAya MaruyamaAya marked this pull request as ready for review June 6, 2023 09:34
@MaruyamaAya MaruyamaAya added the example example-related issuer or pull request label Jun 6, 2023
@FrankLeeeee FrankLeeeee merged commit b306cec into hpcaitech:main Jun 7, 2023
ver217 pushed a commit to ver217/ColossalAI that referenced this pull request Jul 13, 2023
* Modify torch version requirement to adapt torch 2.0

* modify palm example using new booster API

* roll back

* fix port

* polish

* polish
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

example example-related issuer or pull request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants