generated from mwc/lab_tinylm
Applying patch
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user