ShishirPatil/gorilla

[Apibench] Resume interrupted LLM generations from last generation

Open

#484 opened on Jun 24, 2024

 (1 comment) (0 reactions) (0 assignees)Python (1,397 forks)batch import
apibench-datagood first issue

Repository metrics

Stars
 (12,996 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

Describe the issue When we try to launch LLM generations on APIBench, if the generation is interrupted midway, currently the partial file is deleted and restart generation from start. What we should instead do is check where the generation stopped and resume from the last generated index. This is specially critical for large models where generation can take >1 hours.

ID datapoint 2. Provider: This affects all 3: TorchHub/HuggingFace/PyTorch Hub

File that needs to be edited: https://github.com/ShishirPatil/gorilla/blob/main/eval/get_llm_responses.py

Contributor guide