Fix temperature in demo

README.md

@@ -93,8 +93,8 @@ python ./demo/demo_mlx.py \
     --hardware ./demo/hardware/c4dm-disklavier.json \
     --midi_control_signal 67 \
     --midi_reset_control_signal 66 \
-    --temp 0.9 \
-    --min_p 0.03
+    --temp 0.85 \
+    --min_p 0.05
 ```
 
 A MIDI input device is not strictly required to play around with the demo: By using the `--midi_path` and `--midi_through` arguments you can mock real-time input by playing from a MIDI file. All that is required are MIDI drivers (e.g., CoreMIDI) and a virtual software instrument (e.g., Fluidsynth, Pianoteq) to render the output. In this mode, you can initiate the model takeover by pressing the enter key.
@@ -107,7 +107,7 @@ python ./demo/demo_mlx.py \
     --midi_path ${MIDI_PATH} \
     --midi_through <midi-playback-port> \
     --midi_out <midi-playback-port> \
-    --temp 0.9 \
+    --temp 0.85 \
     --min_p 0.05
 ```
 

demo/demo_mlx.py

@@ -696,7 +696,7 @@ def decode_tokens(
             logits[:, tokenizer.tok_to_id[tokenizer.eos_tok]] = float("-inf")
 
         if temperature > 0.0:
-            next_token_ids = sample_min_p(logits, min_p).flatten()
+            next_token_ids = sample_min_p(logits / temperature, min_p).flatten()
         else:
             next_token_ids = mx.argmax(logits, axis=-1).flatten()