Zhouyan248 commited on
Commit
862185b
1 Parent(s): e45b743

Update base/app.py

Browse files
Files changed (1) hide show
  1. base/app.py +2 -3
base/app.py CHANGED
@@ -8,7 +8,6 @@ import cv2
8
  import pandas as pd
9
  import torchvision
10
  import random
11
- from huggingface_hub import snapshot_download
12
  from models import get_models
13
 
14
  from pipelines.pipeline_videogen import VideoGenPipeline
@@ -16,8 +15,8 @@ from download import find_model
16
  from diffusers.schedulers import DDIMScheduler, DDPMScheduler, PNDMScheduler, EulerDiscreteScheduler
17
  from diffusers.models import AutoencoderKL
18
  from transformers import CLIPTokenizer, CLIPTextModel, CLIPTextModelWithProjection
19
- from omegaconf import OmegaConf
20
- from models import get_models
21
  config_path = "./base/configs/sample.yaml"
22
  args = OmegaConf.load("./base/configs/sample.yaml")
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
8
  import pandas as pd
9
  import torchvision
10
  import random
 
11
  from models import get_models
12
 
13
  from pipelines.pipeline_videogen import VideoGenPipeline
 
15
  from diffusers.schedulers import DDIMScheduler, DDPMScheduler, PNDMScheduler, EulerDiscreteScheduler
16
  from diffusers.models import AutoencoderKL
17
  from transformers import CLIPTokenizer, CLIPTextModel, CLIPTextModelWithProjection
18
+
19
+
20
  config_path = "./base/configs/sample.yaml"
21
  args = OmegaConf.load("./base/configs/sample.yaml")
22
  device = "cuda" if torch.cuda.is_available() else "cpu"