Update app.py

app.py

@@ -8,15 +8,16 @@ import random
 
 from skyreelsinfer.offload import Offload, OffloadConfig
 from skyreelsinfer.pipelines import SkyreelsVideoPipeline
-
 from skyreelsinfer import TaskType
-
 #from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
 
+from diffusers import HunyuanVideoTransformer3DModel
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
 from PIL import Image
 
+from transformers import LlamaModel
+
 import torch
 
 torch.backends.cuda.matmul.allow_tf32 = False
@@ -29,6 +30,8 @@ torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 
+torch.backends.cuda.enable_cudnn_sdp(False) # Still a good idea to keep it.
+
 os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
 os.environ["SAFETENSORS_FAST_GPU"] = "1"