feat(optimizer): fine-grained options & presets

pykeio · Apr 16, 2023 · d907f0b · d907f0b
1 parent 7b4c205
commit d907f0b
Show file tree

Hide file tree

Showing 3 changed files with 193 additions and 518 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -12,6 +12,7 @@ scipy~=1.10
 spacy~=3.5
 termcolor>=2,<3
 tomli-w
+tomli
 torch>1,<2 # torch.onnx.export is a bit broken with torch 2
 transformers>=4.21.0,<5
 yaspin~=2.3