Update app.py

app.py

@@ -3,20 +3,20 @@ import torch
 import os
 from diffusers import AutoencoderKLLTXVideo, LTXImageToVideoPipeline, LTXVideoTransformer3DModel
 from transformers import T5EncoderModel
-from diffusers.utils import export_to_video, load_image #, PIL_INTERPOLATION
+from diffusers.utils import export_to_video #, load_image #, PIL_INTERPOLATION
 
 import gradio as gr
 import numpy as np
 import random
 from PIL import Image
-import imageio.v3
+# import imageio.v3
 
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
 torch.backends.cudnn.allow_tf32 = False
 torch.backends.cudnn.deterministic = False
-torch.backends.cudnn.benchmark = False
+torch.backends.cudnn.benchmark = True
 #torch.backends.cuda.preferred_blas_library="cublas"
 #torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
@@ -80,17 +80,17 @@ iface = gr.Interface(
     inputs=[
         gr.Image(type="filepath", label="Image"),
         gr.Textbox(lines=2, label="Prompt"),
-        gr.Textbox(lines=2, label="Negative Prompt"),
+        gr.Textbox(lines=2, label="Negative Prompt", value="blurry, distorted"),
         gr.Slider(minimum=256, maximum=1024, step=8, value=704, label="Width"),
         gr.Slider(minimum=256, maximum=1024, step=8, value=704, label="Height"),
         gr.Slider(minimum=16, maximum=256, step=16, value=111, label="Number of Frames"),
-        gr.Slider(minimum=0.0, maximum=30.0, step=0.01, value=3.8, label="Guidance Scale"),
-        gr.Slider(minimum=1, maximum=100, step=1, value=40, label="Number of Inference Steps"),
+        gr.Slider(minimum=0.0, maximum=30.0, step=0.05, value=3.35, label="Guidance Scale"),
+        gr.Slider(minimum=1, maximum=100, step=1, value=50, label="Number of Inference Steps"),
         gr.Slider(minimum=1, maximum=60, step=1, value=25, label="FPS"),
     ],
     outputs=gr.Video(label="Generated Video"),
     title="LTX-Video Test D",
-    description="Generate video from image with LTX-Video.",
+    description="Generate video from image with LTX-Image-to-Video.",
 )
 
 iface.launch()