mwc/lab_tinylm
Template
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add envrc for automatic installation

Browse Source
This commit is contained in:
chris
2026-02-27 16:21:58 -05:00
parent 01f58ded9a
commit 0a37d36bbd
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.envrc Normal file
View File

@@ -0,0 +1 @@
source .venv/bin/activate

2
tlm/cli.py
View File

@@ -51,7 +51,7 @@ def generate(length, context_window_words, filepath, gutenberg, list_gutenberg,
corpus.extend(tokenize_text(mail_text, tokenize_opts)) corpus.extend(tokenize_text(mail_text, tokenize_opts))
if not corpus: if not corpus:
raise click.UsageError("Must specify at least one of --text, --gutenberg, or --mbox for training data.") raise click.UsageError("No training data provided. Must specify at least one of --filepath, --gutenberg, or --mbox.")
# Train and generate # Train and generate
model = TinyLanguageModel(n=context_window_words) model = TinyLanguageModel(n=context_window_words)