Spaces:
Running
Running
Ruslan Magana Vsevolodovna
commited on
Commit
·
50b2d05
1
Parent(s):
4b68d40
Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def get_output_video(text):
|
|
62 |
models_root=models_root,
|
63 |
is_reusable=True,
|
64 |
is_verbose=True,
|
65 |
-
dtype=torch.float16 if fp16 else torch.float32
|
66 |
device='cuda' #'cpu'
|
67 |
)
|
68 |
log_gpu_memory()
|
|
|
62 |
models_root=models_root,
|
63 |
is_reusable=True,
|
64 |
is_verbose=True,
|
65 |
+
dtype=torch.float16 if fp16 else torch.float32 ,#param ["float32", "float16", "bfloat16"] #float32 is faster than float16 but uses more GPU memory.
|
66 |
device='cuda' #'cpu'
|
67 |
)
|
68 |
log_gpu_memory()
|