Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add fittable #140

Merged
merged 64 commits into from
Feb 7, 2025
Merged
Changes from 1 commit
Commits
Show all changes
64 commits
Select commit Hold shift + click to select a range
4078a3b
Fix tokenizer issue
stephantul Dec 22, 2024
09f888d
fix issue with warning
stephantul Dec 22, 2024
2167a4e
regenerate lock file
stephantul Dec 22, 2024
c95dca5
fix lock file
stephantul Dec 22, 2024
b5d8bb7
Try to not select 2.5.1
stephantul Dec 22, 2024
3e68669
fix: issue with dividers in utils
stephantul Dec 22, 2024
1ae4d61
Try to not select 2.5.0
stephantul Dec 22, 2024
1349b0c
fix: do not up version
stephantul Dec 22, 2024
4b83d59
Attempt special fix
stephantul Dec 22, 2024
9515b83
merge
stephantul Dec 23, 2024
dfd865b
feat: add training
stephantul Dec 23, 2024
c4ba272
merge with old
stephantul Dec 23, 2024
4713bfa
fix: no grad
stephantul Dec 24, 2024
e8058bb
use numpy
stephantul Dec 24, 2024
a59127e
Add train_test_split
stephantul Dec 24, 2024
310fbb5
fix: issue with fit not resetting
stephantul Dec 24, 2024
b1899d1
feat: add lightning
stephantul Dec 28, 2024
e27f9dc
merge
stephantul Dec 28, 2024
8df3aaf
Fix bugs
stephantul Jan 3, 2025
839d88a
fix: reviewer comments
stephantul Jan 5, 2025
8457357
fix train issue
stephantul Jan 5, 2025
a750709
fix issue with trainer
stephantul Jan 7, 2025
e83c54e
fix: truncate during training
stephantul Jan 7, 2025
803565d
feat: tokenize maximum length truncation
stephantul Jan 7, 2025
9052806
fixes
stephantul Jan 8, 2025
2f9fbf4
typo
stephantul Jan 8, 2025
f1e08c3
Add progressbar
stephantul Jan 8, 2025
bb54a76
small code changes, add docs
stephantul Jan 8, 2025
69ee4ee
fix training comments
stephantul Jan 8, 2025
9962be7
Merge branch 'main' into add-fittable
stephantul Jan 16, 2025
ffec235
Add pipeline saving
stephantul Jan 16, 2025
0af84fc
fix bug
stephantul Jan 16, 2025
c829745
fix issue with normalize test
stephantul Jan 16, 2025
9ce65a1
change default batch size
stephantul Jan 17, 2025
e1169fb
feat: add sklearn skops pipeline
stephantul Jan 20, 2025
f096824
Device handling and automatic batch size
stephantul Jan 20, 2025
ff3ebdf
Add docstrings, defaults
stephantul Jan 20, 2025
b4e966a
docs
stephantul Jan 20, 2025
8f65bfd
fix: rename
stephantul Jan 21, 2025
8cdb668
fix: rename
stephantul Jan 21, 2025
e96a72a
fix installation
stephantul Jan 21, 2025
3e76083
rename
stephantul Jan 21, 2025
9f1cb5a
Add training tutorial
stephantul Jan 23, 2025
e2d92b9
Add tutorial link
stephantul Jan 23, 2025
657cef0
Merge branch 'main' into add-fittable
stephantul Jan 24, 2025
773009f
test: add tests
stephantul Jan 24, 2025
7015341
fix tests
stephantul Jan 24, 2025
8ab8456
tests: fix tests
stephantul Jan 24, 2025
e21e61f
Address comments
stephantul Jan 26, 2025
ff75af9
Add inference reqs to train reqs
stephantul Jan 26, 2025
87de7c4
fix normalize
stephantul Jan 26, 2025
1fb33f1
update lock file
stephantul Jan 26, 2025
59f0076
Merge branch 'main' into add-fittable
stephantul Jan 26, 2025
009342b
Merge branch 'main' into add-fittable
stephantul Feb 3, 2025
261a9b4
fix: move modelcards
stephantul Feb 3, 2025
e1d53ac
fix: batch size
stephantul Feb 3, 2025
6b5f991
update lock file
stephantul Feb 3, 2025
759b96c
Update model2vec/inference/README.md
stephantul Feb 7, 2025
7caf9bc
Update model2vec/inference/README.md
stephantul Feb 7, 2025
c7b68b6
Update model2vec/inference/README.md
stephantul Feb 7, 2025
be7baa1
Update model2vec/train/classifier.py
stephantul Feb 7, 2025
cc74618
fix: encode args
stephantul Feb 7, 2025
a4d8d6c
fix: trust_remote_code
stephantul Feb 7, 2025
a0d56d5
fix notebook
stephantul Feb 7, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
regenerate lock file
stephantul committed Dec 22, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
commit 2167a4e270f50f9f7d6210cc22f83611d0793b08
Loading

Unchanged files with check annotations Beta

tokenizer_data["model"]["vocab"] = reindexed
elif model_type == "Unigram":
logger.warning("Removing tokens from a unigram tokenizer is not supported.")
return tokenizer

Check warning on line 72 in model2vec/distill/tokenizer.py

Codecov / codecov/patch

model2vec/distill/tokenizer.py#L71-L72

Added lines #L71 - L72 were not covered by tests
elif model_type == "BPE":
logger.warning("Removing tokens from a BPE tokenizer is not supported.")
return tokenizer

Check warning on line 76 in model2vec/distill/tokenizer.py

Codecov / codecov/patch

model2vec/distill/tokenizer.py#L75-L76

Added lines #L75 - L76 were not covered by tests
else:
raise ValueError(f"Unknown model type {model_type}")