Skip to content

fix: Rework memory handling for rustmark - #50

Merged
jcollomosse merged 5 commits into
mainfrom
mathern/memory-rsutmark-rework
Nov 6, 2026
Merged

jcollomosse merged 5 commits into
mainfrom
mathern/memory-rsutmark-rework

Conversation

@tmathern

@tmathern tmathern commented Nov 5, 2026 •

Copy link
Copy Markdown
Member

Description

  • Reduces Rust's watermarking memory use by reusing model input and avoiding full-image copies.
  • Adds empty-image handling (fixes a potential panic).

Related Issue

N/A

Motivation and Context

Reducing memory usage.

How Has This Been Tested?

  • (Temporary) probes ensuring watermark parity between this branch and main.
  • Unit tests.

Screenshots (if appropriate): N/A

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

  • I have signed the Adobe Open Source CLA.
  • My code follows the code style of this project.
  • My change requires a change to the documentation.
  • I have updated the documentation accordingly.
  • I have read the CONTRIBUTING document.
  • I have added tests to cover my changes.
  • All new and existing tests passed.

@tmathern tmathern self-assigned this Nov 5, 2026
@tmathern
tmathern marked this pull request as ready for review November 5, 2026 23:51
@jcollomosse
jcollomosse merged commit b7f227a into main Nov 6, 2026
8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants