mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
call GC between each e2e test to avoid OOM
This commit is contained in:
parent
5a8389ee0d
commit
19bc081658
|
@ -1,3 +1,4 @@
|
||||||
|
import gc
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Iterator
|
from typing import Iterator
|
||||||
from warnings import warn
|
from warnings import warn
|
||||||
|
@ -30,6 +31,13 @@ from refiners.foundationals.latent_diffusion.stable_diffusion_xl.model import St
|
||||||
from tests.utils import ensure_similar_images
|
from tests.utils import ensure_similar_images
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def ensure_gc():
|
||||||
|
# Avoid GPU OOMs
|
||||||
|
# See https://github.com/pytest-dev/pytest/discussions/8153#discussioncomment-214812
|
||||||
|
gc.collect()
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope="module")
|
@pytest.fixture(scope="module")
|
||||||
def ref_path(test_e2e_path: Path) -> Path:
|
def ref_path(test_e2e_path: Path) -> Path:
|
||||||
return test_e2e_path / "test_diffusion_ref"
|
return test_e2e_path / "test_diffusion_ref"
|
||||||
|
|
Loading…
Reference in a new issue