1. 06 Apr, 2023 7 commits
    • YY Lin's avatar
      [Chat]Add Peft support & fix the ptx bug (#3433) · 62f4e2eb
      YY Lin authored
      * Update ppo.py
      
      Fix the bug of fetching wrong batch data
      
      * Add peft model support in SFT and Prompts training
      
      In stage-1 and stage-3, the peft model supports are added. So the trained artifacts will be only a small lora additions instead of the whole bunch of files.
      
      * Delete test_prompts.txt
      
      * Delete test_pretrained.txt
      
      * Move the peft stuffs to a community folder.
      
      * Move the demo sft to community
      
      * delete dirty files
      
      * Add instructions to install peft using source
      
      * Remove Chinese comments
      
      * remove the Chinese comments
      62f4e2eb
    • Dr-Corgi's avatar
      [chat]fix save_model(#3377) · 73afb635
      Dr-Corgi authored
      The function save_model should be a part of PPOTrainer.
      73afb635
    • kingkingofall's avatar
      [chat]fix readme (#3429) · 57a3c4db
      kingkingofall authored
      * fix stage 2
      
      fix stage 2
      
      * add torch
      57a3c4db
    • Frank Lee's avatar
    • YH's avatar
      Fix typo (#3448) · 8f740deb
      YH authored
      8f740deb
    • ver217's avatar
      [test] reorganize zero/gemini tests (#3445) · 933048ad
      ver217 authored
      933048ad
    • Camille Zhong's avatar
      [Chat] fix the tokenizer "int too big to convert" error in SFT training (#3453) · 72cb4dd4
      Camille Zhong authored
      * Add RoBERTa for RLHF Stage 2 & 3 (test)
      
      RoBERTa for RLHF Stage 2 & 3 (still in testing)
      
      * Revert "Add RoBERTa for RLHF Stage 2 & 3 (test)"
      
      This reverts commit 06741d894dcbe958acd4e10d771f22275e20e368.
      
      * Add RoBERTa for RLHF stage 2 & 3
      
      1. add roberta folder under model folder
      2. add  roberta option in train_reward_model.py
      3. add some test in testci
      
      * Update test_ci.sh
      
      * Revert "Update test_ci.sh"
      
      This reverts commit 9c7352b81766f3177d31eeec0ec178a301df966a.
      
      * Add RoBERTa for RLHF Stage 2 & 3 (test)
      
      RoBERTa for RLHF Stage 2 & 3 (still in testing)
      
      * Revert "Add RoBERTa for RLHF Stage 2 & 3 (test)"
      
      This reverts commit 06741d894dcbe958acd4e10d771f22275e20e368.
      
      * Add RoBERTa for RLHF stage 2 & 3
      
      1. add roberta folder under model folder
      2. add  roberta option in train_reward_model.py
      3. add some test in testci
      
      * Update test_ci.sh
      
      * Revert "Update test_ci.sh"
      
      This reverts commit 9c7352b81766f3177d31eeec0ec178a301df966a.
      
      * update roberta with coati
      
      * chat ci update
      
      * Revert "chat ci update"
      
      This reverts commit 17ae7ae01fa752bd3289fc39069868fde99cf846.
      
      * [Chat] fix the tokenizer "int too big to convert" error in SFT training
      
      fix the tokenizer error during SFT training using Bloom and OPT
      72cb4dd4
  2. 05 Apr, 2023 2 commits
  3. 04 Apr, 2023 6 commits
  4. 03 Apr, 2023 2 commits
    • Frank Lee's avatar
      [test] fixed gemini plugin test (#3411) · 638a07a7
      Frank Lee authored
      * [test] fixed gemini plugin test
      
      * polish code
      
      * polish code
      638a07a7
    • Camille Zhong's avatar
      [chatgpt] add pre-trained model RoBERTa for RLHF stage 2 & 3 (#3223) · 30412866
      Camille Zhong authored
      * Add RoBERTa for RLHF Stage 2 & 3 (test)
      
      RoBERTa for RLHF Stage 2 & 3 (still in testing)
      
      * Revert "Add RoBERTa for RLHF Stage 2 & 3 (test)"
      
      This reverts commit 06741d894dcbe958acd4e10d771f22275e20e368.
      
      * Add RoBERTa for RLHF stage 2 & 3
      
      1. add roberta folder under model folder
      2. add  roberta option in train_reward_model.py
      3. add some test in testci
      
      * add test for reward model training
      
      * Update test_ci.sh
      
      * Revert "Update test_ci.sh"
      
      This reverts commit 9c7352b81766f3177d31eeec0ec178a301df966a.
      
      * Add RoBERTa for RLHF Stage 2 & 3 (test)
      
      RoBERTa for RLHF Stage 2 & 3 (still in testing)
      
      * Revert "Add RoBERTa for RLHF Stage 2 & 3 (test)"
      
      This reverts commit 06741d894dcbe958acd4e10d771f22275e20e368.
      
      * Add RoBERTa for RLHF stage 2 & 3
      
      1. add roberta folder under model folder
      2. add  roberta option in train_reward_model.py
      3. add some test in testci
      
      * Update test_ci.sh
      
      * Revert "Update test_ci.sh"
      
      This reverts commit 9c7352b81766f3177d31eeec0ec178a301df966a.
      
      * update roberta with coati
      30412866
  5. 02 Apr, 2023 1 commit
  6. 01 Apr, 2023 1 commit
    • Jan Roudaut's avatar
      [doc] polish diffusion example (#3386) · dd367ce7
      Jan Roudaut authored
      * [examples/images/diffusion]: README.md: typo fixes
      
      * Update README.md
      
      * Grammar fixes
      
      * Reformulated "Step 3" (xformers) introduction
      
      to the cost => at the cost + reworded pip availability.
      dd367ce7
  7. 31 Mar, 2023 3 commits
    • Jan Roudaut's avatar
      Typofix: malformed `xformers` version (#3384) · 51cd2fec
      Jan Roudaut authored
      s/0.12.0/0.0.12/
      51cd2fec
    • ver217's avatar
      [booster] implement Gemini plugin (#3352) · 5f2e34e6
      ver217 authored
      * [booster] add gemini plugin
      
      * [booster] update docstr
      
      * [booster] gemini plugin add coloparam convertor
      
      * [booster] fix coloparam convertor
      
      * [booster] fix gemini plugin device
      
      * [booster] add gemini plugin test
      
      * [booster] gemini plugin ignore sync bn
      
      * [booster] skip some model
      
      * [booster] skip some model
      
      * [booster] modify test world size
      
      * [booster] modify test world size
      
      * [booster] skip test
      5f2e34e6
    • HELSON's avatar
      [moe] add checkpoint for moe models (#3354) · 1a1d68b0
      HELSON authored
      * [moe] add checkpoint for moe models
      
      * [hotfix] fix bugs in unit test
      1a1d68b0
  8. 30 Mar, 2023 7 commits
  9. 29 Mar, 2023 11 commits