Spaces:
Runtime error
Runtime error
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()
|