ShishirPatil/gorilla

[Apibench] Resume interrupted LLM generations from last generation

Offen

#484 geöffnet am 24.06.2024

 (1 Kommentar) (0 Reaktionen) (0 zugewiesene Personen)Python (1.399 Forks)batch import
apibench-datagood first issue

Repository-Metriken

Stars
 (12.997 Sterne)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

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