* 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:
lewtun
2024-03-01 17:29:42 +01:00
committed by GitHub
co-authored by Alvaro Bartolome Philipp Schmid
parent d17fd7cd3b
commit ff618a4d13
10 changed files with 150 additions and 25 deletions
+2 -2
View File
@@ -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",