From d8bf1beabe39bcecadaecdb538a56d6cbc6afced Mon Sep 17 00:00:00 2001 From: Will Lin Date: Tue, 6 May 2025 17:56:49 -0700 Subject: [PATCH] fix test --- fastvideo/v1/tests/transformers/test_flux.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/fastvideo/v1/tests/transformers/test_flux.py b/fastvideo/v1/tests/transformers/test_flux.py index d0511f3aa..80532cbdf 100644 --- a/fastvideo/v1/tests/transformers/test_flux.py +++ b/fastvideo/v1/tests/transformers/test_flux.py @@ -18,13 +18,14 @@ logger = init_logger(__name__) os.environ["MASTER_ADDR"] = "localhost" os.environ["MASTER_PORT"] = "29503" -# BASE_MODEL_PATH = "black-forest-labs/FLUX.1-dev" -BASE_MODEL_PATH = "/home/test/.cache/huggingface/hub/models--black-forest-labs--FLUX.1-dev/snapshots/0ef5fff789c832c5c7f4e127f94c8b54bbcced44/" +BASE_MODEL_PATH = "black-forest-labs/FLUX.1-dev" +# BASE_MODEL_PATH = "/home/test/.cache/huggingface/hub/models--black-forest-labs--FLUX.1-dev/snapshots/0ef5fff789c832c5c7f4e127f94c8b54bbcced44/" -# MODEL_PATH = maybe_download_model(BASE_MODEL_PATH, -# local_dir=os.path.join( -# 'data', BASE_MODEL_PATH)) -TRANSFORMER_PATH = os.path.join(BASE_MODEL_PATH, "transformer") + +MODEL_PATH = maybe_download_model(BASE_MODEL_PATH, + local_dir=os.path.join( + 'data', BASE_MODEL_PATH)) +TRANSFORMER_PATH = os.path.join(MODEL_PATH, "transformer") @pytest.mark.usefixtures("distributed_setup")