yyk19 commited on
Commit
b928f57
1 Parent(s): 7508c02

save memory while using cuda.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ def load_ckpt(model_ckpt = "LAION-Glyph-10M-Epoch-5"):
106
  allow_run_generation = False
107
  return output_str, None, allow_run_generation
108
 
109
- SAVE_MEMORY = False
110
  disable_verbosity()
111
  if SAVE_MEMORY:
112
  enable_sliced_attention()
 
106
  allow_run_generation = False
107
  return output_str, None, allow_run_generation
108
 
109
+ SAVE_MEMORY = True #False
110
  disable_verbosity()
111
  if SAVE_MEMORY:
112
  enable_sliced_attention()