mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-08-17 11:15:00 +08:00
🪁 (#129)
* Add Gemma 7B recipe * Use Gemma template * Make it work for dolly lol * Enable cahce * Clean up * DPO to the max * DPO, DPO, DPO * Add openhermes * Add custom configs * Add kwargs * Fix config * Bump deps * Move old recipes * Add doc * Add norte * Renable cache * Nuke * Clean * Apply suggestions from code review Co-authored-by: Alvaro Bartolome <alvaro@argilla.io> * Fix isort * Update README.md * Update config_full.yaml --------- Co-authored-by: Alvaro Bartolome <alvaro@argilla.io> Co-authored-by: Philipp Schmid <32632186+philschmid@users.noreply.github.com>
This commit is contained in:
co-authored by
Alvaro Bartolome
Philipp Schmid
parent
d17fd7cd3b
commit
ff618a4d13
@@ -41,7 +41,7 @@ if stale_egg_info.exists():
|
||||
# IMPORTANT: all dependencies should be listed here with their version requirements, if any.
|
||||
# * If a dependency is fast-moving (e.g. transformers), pin to the exact version
|
||||
_deps = [
|
||||
"accelerate==0.23.0",
|
||||
"accelerate==0.27.2",
|
||||
"bitsandbytes==0.41.2.post2",
|
||||
"black==23.1.0",
|
||||
"datasets==2.14.6",
|
||||
@@ -65,7 +65,7 @@ _deps = [
|
||||
"scipy",
|
||||
"tensorboard",
|
||||
"torch==2.1.2",
|
||||
"transformers==4.36.2",
|
||||
"transformers>=4.38.2", # Fixes RoPE computation
|
||||
"trl==0.7.10",
|
||||
"jinja2>=3.0.0",
|
||||
"tqdm>=4.64.1",
|
||||
|
||||
Reference in New Issue
Block a user