Update generativevideo2.py

generativevideo2.py

@@ -12,14 +12,6 @@ from typing import Any
 import time
 from contextlib import contextmanager
 from gradio.helpers import create_examples 
-from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
-
-quantization_config = BitsAndBytesConfig(
-    load_in_4bit=True,
-    bnb_4bit_compute_dtype=torch.float16, # Or torch.float16
-    bnb_4bit_use_double_quant=True,
-    bnb_4bit_quant_type="nf4"
-)
 
 @contextmanager
 def timer(name: str):
@@ -134,6 +126,8 @@ current_dir = Path(__file__).parent
 sys.path.insert(0, str(current_dir / "packages" / "ltx-pipelines" / "src"))
 sys.path.insert(0, str(current_dir / "packages" / "ltx-core" / "src"))
 
+import spaces
+import flash_attn_interface
 import time
 import gradio as gr
 import numpy as np
@@ -143,7 +137,6 @@ from typing import Optional
 from pathlib import Path
 import torchaudio
 from huggingface_hub import hf_hub_download, snapshot_download
-import spaces
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_core.model.video_vae import TilingConfig
 from ltx_core.model.audio_vae.ops import AudioProcessor