LiquidoNoNewtoniano commited on
Commit
7b7e408
1 Parent(s): a5ba672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained("nlpconnect/vit-gpt2-image-captioning"
12
 
13
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
14
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
15
- pipe.enable_model_cpu_offload()
16
 
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
  model.to(device)
 
12
 
13
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
14
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
15
+
16
 
17
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
18
  model.to(device)