Update README.md

README.md

@@ -51,6 +51,19 @@ response = model.generate(
 )
 print(response)
 ```
+**Generation Parameters**:
+- wav_path (`str`) -- Path to the audio file (format = wav, flac, mp3, etc as long as `soundfile.read` supports)
+- prompt (`str`) -- Text input to the model
+- prompt_pattern (`str`) -- Chat template that is augmented with special tokens, and it must be set the same as one during training
+- max_length (`int`, *optional*, defaults to 150)
+- num_beams (`int`, *optional*, defaults to 4)
+- do_sample (`bool`, *optional*, defaults to True)
+- min_length (`int`, *optional*, defaults to 1)
+- top_p (`float`, *optional*, defaults to 0.9)
+- repetition_penalty (`float`, *optional*, defaults to 1.0),
+- length_penalty (`float`, *optional*, defaults to 1.0),
+- temperature (`float`, *optional*, defaults to 1.0),
+- streamer (`TextIteratorStreamer`, *optional*, defaults to `None`) -- this allows streaming output
 
 ## Evaluation Results