Zhouyan248 commited on
Commit
a7a77e5
β€’
1 Parent(s): d5be5ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -19
app.py CHANGED
@@ -12,25 +12,6 @@ config_path = "./configs/sample_i2v.yaml"
12
  args = OmegaConf.load(config_path)
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
- # ------- get model ---------------
16
- # model_i2V = model_i2v_fun()
17
- # model_i2V.to("cuda")
18
-
19
- # vae, model, text_encoder, diffusion = model_i2v_fun(args)
20
- # vae.to(device)
21
- # model.to(device)
22
- # text_encoder.to(device)
23
-
24
- # if args.use_fp16:
25
- # vae.to(dtype=torch.float16)
26
- # model.to(dtype=torch.float16)
27
- # text_encoder.to(dtype=torch.float16)
28
-
29
- # if args.enable_xformers_memory_efficient_attention and device=="cuda":
30
- # if is_xformers_available():
31
- # model.enable_xformers_memory_efficient_attention()
32
- # else:
33
- # raise ValueError("xformers is not available. Make sure it is installed correctly")
34
 
35
 
36
  css = """
 
12
  args = OmegaConf.load(config_path)
13
  device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
 
16
 
17
  css = """