UMass-Rescue / UMass-Rescue/RescueBox

Plugin: Batch processing using LLMs in text-summary and image-summary plugin

Open
#92 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

hackathon
Dominant language
Python
Stars
6
Forks
15
Avg merge
14h 42m
Merged PRs (30d)
7

Description

We currently use Ollama (https://ollama.com/) to prompt LLMs in the text-summary and image-summary plugins. Ollama is easy to use and offers a wide variety of LLMs that can be easily downloaded. However, Ollama does not support batch processing of inputs. It always runs one prompt at a time through the LLM. Some other LLM libraries such as vLLM (https://github.com/vllm-project/vllm) supports batch processing of inputs. But the downside with vLLM is that if we need to download popular local models such as Llama 3.1, Gemma3, etc. from Huggingface, these are "gated" models and require Huggingface authentication (using huggingface-cli) to access the gated models.

The task is to both implement batch processing of prompts through an LLM (must support multi-modal models as well), which enabling us to ship the code in an easy to use manner (We can't ask users to provide huggingface login info, for example, at runtime). We also can't ship the models within the installer (thus eliminating the need for runtime downloads through Huggingface) because that would make the installer too large.

An easier task would be to implement batching using vLLM or other LLM libraries and report the performance gains achieved through batch processing vs using Ollama.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by locating the text-summary and image-summary plugin entry points and reading how they call Ollama. Compare a batching-capable library such as vLLM against the current single-prompt flow, including multimodal inputs and model distribution constraints. Done should include an easy-to-ship approach and measured performance gains versus Ollama.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.