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

Skip tests that use the meta device - they fail on the MacOS CI runners.

Browse Source
This commit is contained in:
Ryan Dick
2024-10-22 15:56:49 +00:00
parent e545f18a45
commit 740f6eb19f
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/backend/flux/ip_adapter/test_xlabs_ip_adapter_flux.py
View File

@@ -1,4 +1,7 @@
import sys
import accelerate
import pytest
import torch
from invokeai.backend.flux.ip_adapter.state_dict_utils import (
@@ -18,6 +21,7 @@ def test_is_state_dict_xlabs_ip_adapter():
assert is_state_dict_xlabs_ip_adapter(sd)
@pytest.mark.skipif(sys.platform == "darwin", reason="Skipping on macOS")
def test_infer_xlabs_ip_adapter_params_from_state_dict():
# Construct a dummy state_dict with tensors of the correct shape on the meta device.
with torch.device("meta"):
@@ -31,6 +35,7 @@ def test_infer_xlabs_ip_adapter_params_from_state_dict():
assert params.clip_embeddings_dim == 768
@pytest.mark.skipif(sys.platform == "darwin", reason="Skipping on macOS")
def test_initialize_xlabs_ip_adapter_flux_from_state_dict():
# Construct a dummy state_dict with tensors of the correct shape on the meta device.
with torch.device("meta"):