Update app.py
Browse files
app.py
CHANGED
|
@@ -41,9 +41,7 @@ frame_asr = FrameBatchMultiTaskAED(
|
|
| 41 |
|
| 42 |
amp_dtype = torch.float16
|
| 43 |
|
| 44 |
-
llm_pipeline = transformers.pipeline(
|
| 45 |
-
"text-generation", model="meta-llama/Meta-Llama-3-8B", model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
|
| 46 |
-
)
|
| 47 |
|
| 48 |
def convert_audio(audio_filepath, tmpdir, utt_id):
|
| 49 |
"""
|
|
|
|
| 41 |
|
| 42 |
amp_dtype = torch.float16
|
| 43 |
|
| 44 |
+
llm_pipeline = transformers.pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct")
|
|
|
|
|
|
|
| 45 |
|
| 46 |
def convert_audio(audio_filepath, tmpdir, utt_id):
|
| 47 |
"""
|