ghostai1 commited on
Commit
9af771e
·
verified ·
1 Parent(s): bbc42b6

Update barks.py

Browse files
Files changed (1) hide show
  1. barks.py +1 -5
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) 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,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 {