sanchit-gandhi HF staff commited on
Commit
7e248e7
1 Parent(s): 16e998b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ device = "cuda:0"
9
  model.to(device)
10
 
11
  sampling_rate = model.audio_encoder.config.sampling_rate
 
12
 
13
 
14
  def generate_audio(prompt, negative_prompt, guidance_scale=3, seed=0):
 
9
  model.to(device)
10
 
11
  sampling_rate = model.audio_encoder.config.sampling_rate
12
+ text_encoder = model.get_text_encoder()
13
 
14
 
15
  def generate_audio(prompt, negative_prompt, guidance_scale=3, seed=0):