r3gm commited on
Commit
c9344ed
·
verified ·
1 Parent(s): 895163f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -40,9 +40,9 @@ os.environ["TOKENIZERS_PARALLELISM"] = "true"
40
  warnings.filterwarnings("ignore")
41
  IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
42
 
43
- if IS_ZERO_GPU:
44
- print("Loading...")
45
- subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
46
 
47
  # --- FRAME EXTRACTION JS & LOGIC ---
48
 
@@ -318,11 +318,11 @@ for i, lora in enumerate(LORA_MODELS):
318
  print("Failed LoRA:", name_high_tr)
319
  pipe.unload_lora_weights()
320
 
321
- if os.path.exists(CACHE_DIR):
322
- shutil.rmtree(CACHE_DIR)
323
- print("Deleted Hugging Face cache.")
324
- else:
325
- print("No hub cache found.")
326
 
327
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
328
  torch._dynamo.reset()
 
40
  warnings.filterwarnings("ignore")
41
  IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
42
 
43
+ # if IS_ZERO_GPU:
44
+ # print("Loading...")
45
+ # subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
46
 
47
  # --- FRAME EXTRACTION JS & LOGIC ---
48
 
 
318
  print("Failed LoRA:", name_high_tr)
319
  pipe.unload_lora_weights()
320
 
321
+ # if os.path.exists(CACHE_DIR):
322
+ # shutil.rmtree(CACHE_DIR)
323
+ # print("Deleted Hugging Face cache.")
324
+ # else:
325
+ # print("No hub cache found.")
326
 
327
  quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
328
  torch._dynamo.reset()