Ruslan Magana Vsevolodovna commited on
Commit
50b2d05
·
1 Parent(s): 4b68d40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 #param ["float32", "float16", "bfloat16"] #float32 is faster than float16 but uses more GPU memory.
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()