cwhuh commited on
Commit
19a112b
·
1 Parent(s): 09813c0

minor chnages

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -9,6 +9,9 @@ from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_
9
 
10
  from huggingface_hub import hf_hub_download
11
  from safetensors.torch import load_file
 
 
 
12
 
13
  dtype = torch.bfloat16
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
9
 
10
  from huggingface_hub import hf_hub_download
11
  from safetensors.torch import load_file
12
+ import subprocess
13
+
14
+ subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
15
 
16
  dtype = torch.bfloat16
17
  device = "cuda" if torch.cuda.is_available() else "cpu"