Skip to content

Arbiter.predict new API

Choose a tag to compare

@TyTodd TyTodd released this 30 Mar 10:25
· 238 commits to main since this release
7be7fdc

Changed the API for Arbiter.predict

arbiter.predict(input_key1=value, input_key2=value2)

Added batching and resume-ability to VLLMBatchClient

Makes VLLMBatchClient way more tolerant of transient GPU crashes and able to handle larger batch jobs

Added PrecompiledProgram._metadata

Convenience attribute to read the metadata from the README frontmatter on a repo

push_to_hub not takes in extra_files

Allows specifying extra files to push to root of repo.