Update src/transformer.py

src/transformer.py

@@ -7,7 +7,6 @@ from diffusers.models.transformers.transformer_flux import (
     FluxTransformer2DModel,
     Transformer2DModelOutput,
     USE_PEFT_BACKEND,
-    is_torch_version,
     scale_lora_layers,
     unscale_lora_layers,
     logger,
@@ -155,9 +154,6 @@ def tranformer_forward(
 
                 return custom_forward
 
-            ckpt_kwargs: Dict[str, Any] = (
-                {"use_reentrant": False} if is_torch_version(">=", "1.11.0") else {}
-            )
             encoder_hidden_states, hidden_states = torch.utils.checkpoint.checkpoint(
                 create_custom_forward(block),
                 hidden_states,
@@ -204,9 +200,6 @@ def tranformer_forward(
 
                 return custom_forward
 
-            ckpt_kwargs: Dict[str, Any] = (
-                {"use_reentrant": False} if is_torch_version(">=", "1.11.0") else {}
-            )
             hidden_states = torch.utils.checkpoint.checkpoint(
                 create_custom_forward(block),
                 hidden_states,