liguang0115 commited on
Commit
28b109f
·
1 Parent(s): 4bab685

Remove redundant import of torch in app.py and add it back for proper functionality with GPU context and autocasting.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import subprocess
2
  import sys
3
  import spaces
4
-
5
 
6
  @spaces.GPU(duration=10)
7
  @torch.autocast("cuda")
@@ -20,7 +20,7 @@ from functools import partial
20
 
21
  import gradio as gr
22
  import numpy as np
23
- import torch
24
  from omegaconf import OmegaConf
25
  from modeling.pipeline import VMemPipeline
26
  from diffusers.utils import export_to_video
 
1
  import subprocess
2
  import sys
3
  import spaces
4
+ import torch
5
 
6
  @spaces.GPU(duration=10)
7
  @torch.autocast("cuda")
 
20
 
21
  import gradio as gr
22
  import numpy as np
23
+
24
  from omegaconf import OmegaConf
25
  from modeling.pipeline import VMemPipeline
26
  from diffusers.utils import export_to_video