change model video

README.md

@@ -24,15 +24,15 @@ Créez des images et des vidéos de haute qualité avec l'IA grâce aux meilleur
 - Paramètres avancés personnalisables
 - Interface intuitive avec exemples
 
-## 🎬 Text-to-Video (Wan2.1-T2V-14B)
+## 🎬 Text-to-Video (CogVideoX-5B)
 
-**Modèle** : Wan2.1-T2V-14B de Wan-AI (basé sur CogVideoX)
+**Modèle** : CogVideoX-5B de THUDM (Tsinghua University)
 
 ### Fonctionnalités
 - Génération de vidéos depuis du texte
 - Qualité vidéo haute définition (720p)
 - Vidéos de 6 à 10 secondes
-- 14 milliards de paramètres
+- 5 milliards de paramètres
 - Paramètres avancés personnalisables
 
 ## 🚀 Utilisation

video/main.py

@@ -8,10 +8,10 @@ from diffusers.utils import export_to_video
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 
-# Chargement du modèle Wan2.1-T2V-14B
-print("🚀 Chargement du modèle Wan2.1-T2V-14B...")
+# Chargement du modèle CogVideoX-5B
+print("🚀 Chargement du modèle CogVideoX-5B...")
 pipe = CogVideoXPipeline.from_pretrained(
-    "Wan-AI/Wan2.1-T2V-14B",
+    "THUDM/CogVideoX-5b",
     torch_dtype=dtype,
 )
 if torch.cuda.is_available():
@@ -29,7 +29,7 @@ def generate_video(
     progress=gr.Progress(track_tqdm=True)
 ):
     """
-    Génère une vidéo à partir d'un prompt texte avec Wan2.1-T2V-14B
+    Génère une vidéo à partir d'un prompt texte avec CogVideoX-5B
 
     Args:
         prompt: Description de la vidéo à générer
@@ -68,11 +68,11 @@ def generate_video(
 
         # Métadonnées
         metadata = f"""Prompt: {prompt}
-        Modèle: Wan2.1-T2V-14B
-        Steps: {num_inference_steps}
-        Guidance Scale: {guidance_scale}
-        Frames: {num_frames}
-        FPS: 8"""
+Modèle: CogVideoX-5B
+Steps: {num_inference_steps}
+Guidance Scale: {guidance_scale}
+Frames: {num_frames}
+FPS: 8"""
 
         return video_path, metadata
 
@@ -83,10 +83,10 @@ def generate_video(
         raise gr.Error(f"❌ Erreur lors de la génération: {str(e)}")
 
 # Interface Gradio
-with gr.Blocks(theme=gr.themes.Soft(), title="Wan2.1-T2V - Text-to-Video") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="CogVideoX-5B - Text-to-Video") as demo:
     gr.Markdown(
         """
-        # 🎬 Générateur de Vidéos Wan2.1-T2V-14B
+        # 🎬 Générateur de Vidéos CogVideoX-5B
         ### Créez des vidéos à partir de descriptions textuelles
         """
     )
@@ -179,12 +179,12 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Wan2.1-T2V - Text-to-Video") as de
         """
         ---
         ### 📚 À propos
-        Ce générateur utilise **Wan2.1-T2V-14B** de Wan-AI, un modèle de génération vidéo text-to-video basé sur CogVideoX.
+        Ce générateur utilise **CogVideoX-5B** de THUDM (Tsinghua University), un modèle de génération vidéo text-to-video.
 
         **Caractéristiques:**
         - 🎬 Génération de vidéos depuis des descriptions textuelles
-        - 🎨 Qualité vidéo haute définition
-        - ⚡ 14 milliards de paramètres
+        - 🎨 Qualité vidéo haute définition (720p)
+        - ⚡ 5 milliards de paramètres
         - 🎯 Compréhension avancée des prompts
         - 📹 Vidéos de 6 à 10 secondes