okestro-ai-lab
/

FastSLM-ASR

Audio-Text-to-Text

feature-extraction

text-generation

Model card Files Files and versions

speechisalluneed commited on Jan 27

Commit

cf86f4c

·

verified ·

1 Parent(s): e3c686b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -223,7 +223,7 @@ input_ids = tokenizer.apply_chat_template(
     prompt,
     add_generation_prompt=True,
     tokenize=True,
-    reutrn_tensors='pt'
 ).to(model.device)
 # 3. Perform inference

     prompt,
     add_generation_prompt=True,
     tokenize=True,
+    return_tensors='pt'
 ).to(model.device)
 # 3. Perform inference