Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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 |
-
|
| 45 |
-
|
| 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 |
-
|
| 323 |
-
|
| 324 |
-
else:
|
| 325 |
-
|
| 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()
|