[PR #13087] [CLOSED] add retrieval model optimizations #126752

New Issue

GiteaMirror · 2026-05-21T08:03:40-05:00

GiteaMirror commented

2026-05-21 08:03:40 -05:00

📋 Pull Request Information

Original PR: https://github.com/open-webui/open-webui/pull/13087
Author: @chsasank
Created: 4/20/2025
Status: ❌ Closed

Base: main ← Head: sasank

📝 Commits (1)

1001139 add retrieval model optimizations

📊 Changes

2 files changed (+11 additions, -3 deletions)

View changed files

📝 backend/open_webui/retrieval/utils.py (+9 -3)
📝 backend/open_webui/routers/retrieval.py (+2 -0)

📄 Description

Added retrieval model optimizations. torch.cuda.empty_cache() clears up any unnecessary memory occupied by pytorch after inference is complete. Models are loaded in FP16 because that's usually good enough but saves ram by 2x.

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/open-webui/open-webui/pull/13087 **Author:** [@chsasank](https://github.com/chsasank) **Created:** 4/20/2025 **Status:** ❌ Closed **Base:** `main` ← **Head:** `sasank` --- ### 📝 Commits (1) - [`1001139`](https://github.com/open-webui/open-webui/commit/1001139910bd07498ad3882fe6ebbc87c2c53651) add retrieval model optimizations ### 📊 Changes **2 files changed** (+11 additions, -3 deletions) <details> <summary>View changed files</summary> 📝 `backend/open_webui/retrieval/utils.py` (+9 -3) 📝 `backend/open_webui/routers/retrieval.py` (+2 -0) </details> ### 📄 Description Added retrieval model optimizations. `torch.cuda.empty_cache()` clears up any unnecessary memory occupied by pytorch after inference is complete. Models are loaded in FP16 because that's usually good enough but saves ram by 2x. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

GiteaMirror added the pull-request label 2026-05-21 08:03:40 -05:00

GiteaMirror closed this issue

2026-05-21 08:03:43 -05:00

Sign in to join this conversation.

Branches Tags

1 Participants

Notifications

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: github-starred/open-webui#126752