sanchit-gandhi
commited on
Commit
·
22c6f1b
1
Parent(s):
ee252bb
use zero gpu
Browse files
app.py
CHANGED
|
@@ -176,7 +176,7 @@ target_dtype = np.int16
|
|
| 176 |
max_range = np.iinfo(target_dtype).max
|
| 177 |
|
| 178 |
|
| 179 |
-
|
| 180 |
def generate_audio(text_prompt, audio_length_in_s=10.0, play_steps_in_s=2.0, seed=0):
|
| 181 |
max_new_tokens = int(frame_rate * audio_length_in_s)
|
| 182 |
play_steps = int(frame_rate * play_steps_in_s)
|
|
|
|
| 176 |
max_range = np.iinfo(target_dtype).max
|
| 177 |
|
| 178 |
|
| 179 |
+
@spaces.GPU
|
| 180 |
def generate_audio(text_prompt, audio_length_in_s=10.0, play_steps_in_s=2.0, seed=0):
|
| 181 |
max_new_tokens = int(frame_rate * audio_length_in_s)
|
| 182 |
play_steps = int(frame_rate * play_steps_in_s)
|