rahul7star commited on
Commit
61d98ec
·
verified ·
1 Parent(s): 333c9b4

Update app_quant.py

Browse files
Files changed (1) hide show
  1. app_quant.py +2 -0
app_quant.py CHANGED
@@ -3,6 +3,7 @@
3
  Optimized inference for Maya1 + LoRA + SNAC.
4
  Includes side-by-side Base vs LoRA comparison for audio.
5
  """
 
6
 
7
  import gradio as gr
8
  import torch
@@ -136,6 +137,7 @@ def build_maya_prompt(description: str, text: str):
136
  # -------------------------
137
  # Optimized generator
138
  # -------------------------
 
139
  def generate_audio_from_model(model_to_use, description, text, fname="tts.wav"):
140
  logs = []
141
  t0 = time.time()
 
3
  Optimized inference for Maya1 + LoRA + SNAC.
4
  Includes side-by-side Base vs LoRA comparison for audio.
5
  """
6
+ import spaces
7
 
8
  import gradio as gr
9
  import torch
 
137
  # -------------------------
138
  # Optimized generator
139
  # -------------------------
140
+ @spaces.GPU()
141
  def generate_audio_from_model(model_to_use, description, text, fname="tts.wav"):
142
  logs = []
143
  t0 = time.time()