Zhouyan248 commited on
Commit
e45b743
1 Parent(s): 8c4bc8e

Update base/app.py

Browse files
Files changed (1) hide show
  1. base/app.py +7 -0
base/app.py CHANGED
@@ -11,6 +11,13 @@ import random
11
  from huggingface_hub import snapshot_download
12
  from models import get_models
13
 
 
 
 
 
 
 
 
14
  config_path = "./base/configs/sample.yaml"
15
  args = OmegaConf.load("./base/configs/sample.yaml")
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
11
  from huggingface_hub import snapshot_download
12
  from models import get_models
13
 
14
+ from pipelines.pipeline_videogen import VideoGenPipeline
15
+ 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"