Spaces:
Sleeping
Sleeping
sparkleman
commited on
Commit
·
1c4c774
1
Parent(s):
ade4f4a
UPDATE
Browse files
app.py
CHANGED
|
@@ -5,6 +5,9 @@ if os.environ.get("MODELSCOPE_ENVIRONMENT") == "studio":
|
|
| 5 |
|
| 6 |
patch_hub()
|
| 7 |
|
|
|
|
|
|
|
|
|
|
| 8 |
from config import CONFIG, ModelConfig
|
| 9 |
from utils import (
|
| 10 |
cleanMessages,
|
|
|
|
| 5 |
|
| 6 |
patch_hub()
|
| 7 |
|
| 8 |
+
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
| 9 |
+
|
| 10 |
+
|
| 11 |
from config import CONFIG, ModelConfig
|
| 12 |
from utils import (
|
| 13 |
cleanMessages,
|