Add support for jupyter notebooks for training

2025-03-28 16:10:43 +00:00
parent 468ef2ae1f
commit 56242496bb
5 changed files with 343 additions and 11 deletions
--- a/tools/train/setup.py
+++ b/tools/train/setup.py
@@ -15,6 +15,8 @@ setup(
        'datasets>=2.14.6',
        'einops>=0.7.0',
        'flake8>=4.0.0',
+        'ipykernel>=6.0.0', 
+        'ipywidgets>=8.0.0',
        'peft>=0.8.0',
        'peft>=0.8.0',
        'pytest-cov>=4.0.0',