Ruslan Magana Vsevolodovna commited on
Commit
3ad339f
β€’
1 Parent(s): f8e6d87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -47,6 +47,8 @@ def get_output_video(text):
47
  Set the dtype to torch.float16 to save GPU memory.
48
  If you have an Ampere architecture GPU you can use torch.bfloat16.
49
  Set the device to either "cuda" or "cpu". Once everything has finished initializing,
 
 
50
  '''
51
 
52
  def generate_image(
@@ -63,7 +65,7 @@ def get_output_video(text):
63
  models_root=models_root,
64
  is_reusable=True,
65
  is_verbose=True,
66
- dtype=torch.float16 if fp16 else torch.float32 ,#param ["float32", "float16", "bfloat16"] #float32 is faster than float16 but uses more GPU memory.
67
  #device='cuda' #'cpu'
68
  )
69
  #log_gpu_memory()
 
47
  Set the dtype to torch.float16 to save GPU memory.
48
  If you have an Ampere architecture GPU you can use torch.bfloat16.
49
  Set the device to either "cuda" or "cpu". Once everything has finished initializing,
50
+ float32 is faster than float16 but uses more GPU memory.
51
+
52
  '''
53
 
54
  def generate_image(
 
65
  models_root=models_root,
66
  is_reusable=True,
67
  is_verbose=True,
68
+ dtype=torch.float16 if fp16 else torch.float32 ,#param ["float32", "float16", "bfloat16"]
69
  #device='cuda' #'cpu'
70
  )
71
  #log_gpu_memory()