Skip to content

Commit 2c39c46

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Fix params on create_dataset() call.
1 parent 1c3ffba commit 2c39c46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1372,7 +1372,7 @@ def create_dataset(raw_text_sample, tokenizer, sample_expansion_batch_size=10) -
13721372
)
13731373
return dataset
13741374

1375-
phase_i_b_dataset = create_dataset(raw_text_sample=phase_i_b_samples, tokenizer, sample_expansion_batch_size=10)
1375+
phase_i_b_dataset = create_dataset(raw_text_sample=phase_i_b_samples, tokenizer=tokenizer, sample_expansion_batch_size=10)
13761376

13771377
# To Do: Set .fit() params <------<<<
13781378

0 commit comments

Comments
 (0)