helblazer811 commited on
Commit
48daf06
·
1 Parent(s): cac974c

removing gpu offload which is causing storage issues on deployment

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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",