Spaces:
Running
on
Zero
Running
on
Zero
dongyh20
commited on
Commit
·
48f0970
1
Parent(s):
7b80a5d
update space
Browse files
app.py
CHANGED
@@ -1,7 +1,5 @@
|
|
1 |
import os
|
2 |
-
import
|
3 |
-
os.system('nvidia-smi')
|
4 |
-
print(torch.backends.cudnn.version())
|
5 |
os.environ['LOWRES_RESIZE'] = '384x32'
|
6 |
os.environ['HIGHRES_BASE'] = '0x32'
|
7 |
os.environ['VIDEO_RESIZE'] = "0x64"
|
@@ -16,8 +14,6 @@ os.environ['SKIP_LOAD_VIT'] = '1'
|
|
16 |
|
17 |
os.environ["CUDA_LAUNCH_BLOCKING"]='1'
|
18 |
|
19 |
-
import spaces
|
20 |
-
|
21 |
import gradio as gr
|
22 |
import torch
|
23 |
import re
|
|
|
1 |
import os
|
2 |
+
import spaces
|
|
|
|
|
3 |
os.environ['LOWRES_RESIZE'] = '384x32'
|
4 |
os.environ['HIGHRES_BASE'] = '0x32'
|
5 |
os.environ['VIDEO_RESIZE'] = "0x64"
|
|
|
14 |
|
15 |
os.environ["CUDA_LAUNCH_BLOCKING"]='1'
|
16 |
|
|
|
|
|
17 |
import gradio as gr
|
18 |
import torch
|
19 |
import re
|