Ensure deterministic tests can run.

2025-12-06 20:44:33 +08:00 · 2023-05-17 15:16:23 +02:00
1 changed files with 1 additions and 0 deletions
--- a/tests/models/test_models_unet_2d_condition.py
+++ b/tests/models/test_models_unet_2d_condition.py
@@ -40,6 +40,7 @@ from .test_modeling_common import ModelTesterMixin
 logger = logging.get_logger(__name__)
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.use_deterministic_algorithms(True)
+os.environ["CUBLAS_WORKSPACE_CONFIG"] = os.environ.get("CUBLAS_WORKSPACE_CONFIG", ":16:8")


 def create_lora_layers(model, mock_weights: bool = True):