update tests

tests/flux/test_flux_dev.py

@@ -9,7 +9,7 @@ from .utils import run_test
     "height,width,num_inference_steps,attention_impl,cpu_offload,expected_lpips",
     [
         (1024, 1024, 50, "flashattn2", False, 0.139),
-        (2048, 512, 25, "nunchaku-fp16", False, 0.148),
+        (2048, 512, 25, "nunchaku-fp16", False, 0.168),
     ],
 )
 def test_flux_dev(

tests/flux/test_flux_tools.py

@@ -140,5 +140,5 @@ def test_flux_dev_redux():
         attention_impl="nunchaku-fp16",
         cpu_offload=False,
         cache_threshold=0,
-        expected_lpips=(0.143 if get_precision() == "int4" else 0.198),
+        expected_lpips=(0.162 if get_precision() == "int4" else 0.198),
     )

tests/flux/utils.py

@@ -180,7 +180,7 @@ def run_test(
         for lora_name, lora_strength in zip(lora_names, lora_strengths):
             folder_name += f"-{lora_name}_{lora_strength}"
 
-    ref_root = os.path.join("test_results", "ref")
+    ref_root = os.environ.get("NUNCHAKU_TEST_CACHE_ROOT", os.path.join("test_results", "ref"))
     save_dir_16bit = os.path.join(ref_root, dtype_str, model_name, folder_name)
 
     if task in ["t2i", "redux"]: