Spaces:
Sleeping
Sleeping
save memory while using cuda.
Browse files
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()
|