Update barks.py
Browse files
barks.py
CHANGED
@@ -90,7 +90,7 @@ except Exception as e:
|
|
90 |
def print_resource_usage(stage: str):
|
91 |
print(f"--- {stage} ---")
|
92 |
print(f"GPU Memory Allocated: {torch.cuda.memory_allocated() / (1024**3):.2f} GB")
|
93 |
-
print(f"GPU Memory Reserved: {torch.cuda.memory_reserved() / (1024**3):.2f
|
94 |
print(f"CPU Memory Used: {psutil.virtual_memory().percent}%")
|
95 |
print(f"System RAM Available: {psutil.virtual_memory().available / (1024**3):.2f} GB")
|
96 |
print("---------------")
|
@@ -426,10 +426,6 @@ def generate_music(instrumental_prompt: str, vocal_prompt: str, cfg_scale: float
|
|
426 |
finally:
|
427 |
memory_cleanup()
|
428 |
|
429 |
-
# Function to clear inputs
|
430 |
-
def clear_inputs():
|
431 |
-
return "", "", 3.0, 250, 0.9, 1.0, 30, 5, 1000, 120, "none", "none", "none", "none", "none"
|
432 |
-
|
433 |
# 8) CUSTOM CSS
|
434 |
css = """
|
435 |
body {
|
|
|
90 |
def print_resource_usage(stage: str):
|
91 |
print(f"--- {stage} ---")
|
92 |
print(f"GPU Memory Allocated: {torch.cuda.memory_allocated() / (1024**3):.2f} GB")
|
93 |
+
print(f"GPU Memory Reserved: {torch.cuda.memory_reserved() / (1024**3):.2f} GB")
|
94 |
print(f"CPU Memory Used: {psutil.virtual_memory().percent}%")
|
95 |
print(f"System RAM Available: {psutil.virtual_memory().available / (1024**3):.2f} GB")
|
96 |
print("---------------")
|
|
|
426 |
finally:
|
427 |
memory_cleanup()
|
428 |
|
|
|
|
|
|
|
|
|
429 |
# 8) CUSTOM CSS
|
430 |
css = """
|
431 |
body {
|