github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Try to fix test failures affecting MacOS CI runners."

Browse Source 
This reverts commit 216b36c75d.
This commit is contained in:
Ryan Dick
2024-10-09 16:30:40 +00:00
committed by Kent Keirsey
parent b3b7d084d0
commit 118f0ba3bf
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/backend/flux/controlnet/test_state_dict_utils.py
View File

@@ -48,7 +48,8 @@ def test_convert_diffusers_instantx_state_dict_to_bfl_format():
def test_infer_flux_params_from_state_dict():
# Construct a dummy state_dict with tensors of the correct shape on the meta device.
sd = {k: torch.zeros(v, device="meta") for k, v in instantx_sd_shapes.items()}
with torch.device("meta"):
sd = {k: torch.zeros(v) for k, v in instantx_sd_shapes.items()}
sd = convert_diffusers_instantx_state_dict_to_bfl_format(sd)
flux_params = infer_flux_params_from_state_dict(sd)
@@ -69,7 +70,8 @@ def test_infer_flux_params_from_state_dict():
def test_infer_instantx_num_control_modes_from_state_dict():
# Construct a dummy state_dict with tensors of the correct shape on the meta device.
sd = {k: torch.zeros(v, device="meta") for k, v in instantx_sd_shapes.items()}
with torch.device("meta"):
sd = {k: torch.zeros(v) for k, v in instantx_sd_shapes.items()}
sd = convert_diffusers_instantx_state_dict_to_bfl_format(sd)
num_control_modes = infer_instantx_num_control_modes_from_state_dict(sd)
@@ -79,7 +81,8 @@ def test_infer_instantx_num_control_modes_from_state_dict():
def test_load_instantx_from_state_dict():
# Construct a dummy state_dict with tensors of the correct shape on the meta device.
sd = {k: torch.zeros(v, device="meta") for k, v in instantx_sd_shapes.items()}
with torch.device("meta"):
sd = {k: torch.zeros(v) for k, v in instantx_sd_shapes.items()}
sd = convert_diffusers_instantx_state_dict_to_bfl_format(sd)
flux_params = infer_flux_params_from_state_dict(sd)