Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
48daf06
1
Parent(s):
cac974c
removing gpu offload which is causing storage issues on deployment
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ from PIL import Image
|
|
4 |
import math
|
5 |
import io
|
6 |
import base64
|
|
|
|
|
7 |
|
8 |
from concept_attention import ConceptAttentionFluxPipeline
|
9 |
|
@@ -271,6 +273,8 @@ with gr.Blocks(
|
|
271 |
)
|
272 |
|
273 |
if __name__ == "__main__":
|
|
|
|
|
274 |
demo.launch()
|
275 |
# share=True,
|
276 |
# server_name="0.0.0.0",
|
|
|
4 |
import math
|
5 |
import io
|
6 |
import base64
|
7 |
+
import subprocess
|
8 |
+
import os
|
9 |
|
10 |
from concept_attention import ConceptAttentionFluxPipeline
|
11 |
|
|
|
273 |
)
|
274 |
|
275 |
if __name__ == "__main__":
|
276 |
+
if os.path.exists("/data-nvme/zerogpu-offload"):
|
277 |
+
subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
|
278 |
demo.launch()
|
279 |
# share=True,
|
280 |
# server_name="0.0.0.0",
|