Bootstrap Audio Flamingo 3 custom endpoint repo

handler.py

@@ -251,6 +251,7 @@ class EndpointHandler:
                 tokenize=True,
                 add_generation_prompt=True,
                 return_dict=True,
+                return_tensors="pt",
                 audio_kwargs={"sampling_rate": int(sample_rate)},
             )
         except Exception:
@@ -259,6 +260,7 @@ class EndpointHandler:
                 tokenize=True,
                 add_generation_prompt=True,
                 return_dict=True,
+                return_tensors="pt",
             )
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]: