Some basic UCE prompts#2
Open
wlg1 wants to merge 9 commits into
Open
Conversation
LukeSamkharadze
added a commit
that referenced
this pull request
May 28, 2026
Two bugs that compound and break the Steering page's "Shift toward demographic" preset (8 inline pairs + CAA + sdxl_turbo): 1. ActivationsDataloader.renew_buffer raised StopIteration *before* re-materialising the carry-over rows into self.buffer, so the iterate() tail-yield fallback (`if self.buffer is not None: yield self.buffer[self.pointer:]`) silently dropped any under-sized tail. A 13-row train split against batch_size=16 produced zero batches. 2. With the tail now exposed, run_steer.collect_from_loader hit an IndexError. The act loader (flatten=True) treats each tar entry's (seq, dim) activation as `seq` rows; the label loader (flatten=False) treats one entry as one row. PairedLoader zips them per-batch, so an act batch of size 16 paired with a label batch of size 13. Cap iteration to the shorter side — same semantics as the pre-existing zip-truncation behaviour for full-batch runs. Adds two test layers so this regression chain doesn't return: - tests/unit/test_buffer.py — 4 regression tests for ActivationsDataloader and PairedLoader covering the under-sized tail, exact batch boundary, and the common multi-batch case. - tests/integration/test_e2e_cli.py — slow-marked e2e tests that spawn the actual t2i-steer / t2i-localise / t2i-stitch CLIs (same code path as the Streamlit pages) at minimum scale. Caught bug #2 above. Skip on non-MPS so they don't run on CI without an accelerator.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Simple UCE prompts that the paper used for generating race and profession images. can reuse for experiments comparing to baselines