Archives/imaginAIry
2
0
Fork 0
mirror of https://github.com/brycedrennan/imaginAIry synced 2024-11-19 03:25:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #418 from brycedrennan/fix_test_set_spu_full

Browse Source 
fix: configures test_set_gpu_full to run on a m1 mac.
This commit is contained in:
jaydrennan 2023-12-17 23:50:46 -07:00 committed by GitHub
commit 2eee741b20
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_utils/test_model_cache.py
View File

@ -65,10 +65,14 @@ def test_set_cpu_full():
@pytest.mark.skipif(get_device() == "cpu", reason="GPU not available")
def test_set_gpu_full():
device = get_device()
cache = GPUModelCache(
max_cpu_memory_gb=1, max_gpu_memory_gb=0.0000001, device=get_device()
max_cpu_memory_gb=1, max_gpu_memory_gb=0.0000001, device=device
)
assert cache.max_cpu_memory == 1073741824
if device in ("cpu", "mps"):
assert cache.max_cpu_memory == 0
else:
assert cache.max_cpu_memory == 1073741824
model = create_model_of_n_bytes(100_000)
with pytest.raises(RuntimeError):
cache.set("key1", model)