Pass through fast saver latent

This commit is contained in:
2026-07-07 20:01:45 +02:00
parent 4b0b4b6c5a
commit 39997b1d34
2 changed files with 14 additions and 3 deletions
+10
View File
@@ -45,6 +45,16 @@ def test_png_save_writes_matching_latent_sidecar(tmp_path):
assert loaded["keep"] == {"value": 7}
def test_png_save_returns_latent_passthrough(tmp_path):
saver = FastAbsoluteSaver()
latent = {"samples": torch.ones((1, 1, 2, 2))}
result = saver.save_images_fast(**_save_args(tmp_path, latent=latent))
assert result["result"] == (latent,)
assert result["result"][0] is latent
def test_video_save_writes_latent_sidecar_next_to_video(tmp_path):
saver = FastAbsoluteSaver()
latent = {"samples": torch.arange(8, dtype=torch.float32).reshape(2, 1, 2, 2)}