Update app.py
Browse files
app.py
CHANGED
|
@@ -13,6 +13,9 @@ import gradio as gr
|
|
| 13 |
import psutil
|
| 14 |
import time
|
| 15 |
import math
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None)
|
| 18 |
TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None)
|
|
|
|
| 13 |
import psutil
|
| 14 |
import time
|
| 15 |
import math
|
| 16 |
+
from transformers.utils.hub import move_cache
|
| 17 |
+
|
| 18 |
+
move_cache()
|
| 19 |
|
| 20 |
SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None)
|
| 21 |
TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None)
|