Skip to content

Commit 1456a41

Browse files
committed
chore: resolve conflicts in z_image_working_memory test
1 parent fe57a2f commit 1456a41

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/app/invocations/test_z_image_working_memory.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,7 @@ def test_z_image_latents_to_image_requests_working_memory(self, vae_type):
4646
mock_latents = torch.zeros(1, 16, 64, 64)
4747
mock_context.tensors.load.return_value = mock_latents
4848

49-
# Mock the appropriate estimation function
50-
if vae_type == FluxAutoEncoder:
51-
estimation_path = "invokeai.app.invocations.z_image_latents_to_image.estimate_vae_working_memory_flux"
52-
else:
53-
estimation_path = "invokeai.app.invocations.z_image_latents_to_image.estimate_vae_working_memory_sd3"
49+
estimation_path = "invokeai.app.invocations.z_image_latents_to_image.estimate_vae_working_memory_flux"
5450

5551
with patch(estimation_path) as mock_estimate:
5652
expected_memory = 1024 * 1024 * 500 # 500MB

0 commit comments

Comments
 (0)