Skip to content

Commit

Permalink
tweaks
Browse files Browse the repository at this point in the history
  • Loading branch information
Valentin Zulkower committed Nov 15, 2024
1 parent 1562d6a commit 8cf67dd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/handling_large_batches.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
output_folder.mkdir(parents=True, exist_ok=True)

client = GinkgoClient()
queries = MeanEmbeddingQuery.iter_from_fasta(input_file)
queries = MeanEmbeddingQuery.iter_from_fasta(input_file, model="ginkgo-aa0-650m")
for batch_result in client.send_batched_requests(queries, batch_size=100):
for query_result in batch_result:
with open(output_folder / f"{query_result.query_name}.json", "w") as f:
Expand Down

0 comments on commit 8cf67dd

Please sign in to comment.