jauntybrain
/

musicgen-small

Inference Endpoints

Model card Files Files and versions Community

jauntybrain commited on Sep 23, 2023

Commit

9edeb3b

•

1 Parent(s): 180d507

Update handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -34,7 +34,8 @@ class EndpointHandler:
                 outputs = self.model.generate(**inputs, do_sample=False, max_new_tokens=400)
         # postprocess the prediction
-        audio_samples = outputs[0].cpu().numpy()[0].tolist()
         audio_samples = [int(min(max(sample * 32767, -32768), 32767)) for sample in audio_samples]

                 outputs = self.model.generate(**inputs, do_sample=False, max_new_tokens=400)
         # postprocess the prediction
+        sampling_rate = model.config.audio_encoder.sampling_rate
+        audio_samples = audio_values[0].cpu().numpy()[0].tolist()
         audio_samples = [int(min(max(sample * 32767, -32768), 32767)) for sample in audio_samples]