Update app.py

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 import torch
-import torch.distributed as dist
-import torch.multiprocessing as mp
 import time
 import json
 import os
@@ -13,12 +11,8 @@ from src.video_infinity.wrapper import DistWrapper
 def init_pipeline(config):
     pipe = VideoCrafterPipeline.from_pretrained(
         'adamdad/videocrafterv2_diffusers',
-        torch_dtype=torch.float16
+        torch_dtype=torch.float32  # Используем float32 для CPU
     )
-    pipe.enable_model_cpu_offload(
-        gpu_id=config["devices"][dist.get_rank() % len(config["devices"])],
-    )
-    pipe.enable_vae_slicing()
     return pipe
 
 def run_inference(prompt, config):
@@ -48,8 +42,8 @@ def demo(input_text):
         os.makedirs(base_path)
     
     config = {
-        "devices": [0],  # Укажите индексы ваших GPU, например [0] для одной GPU или [0, 1] для двух
-        "base_path": base_path,  # Указываем путь, где будут сохраняться видео
+        "devices": [0],  # Используем CPU
+        "base_path": base_path,  # Указываем путь для сохранения видео
         "pipe_configs": {
             "prompts": [input_text]
         },