Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
5model = PeftModel.from_pretrained(base_model, "kraken07/mistral-7b-spgi")
6tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
7
8# Your input format should match training: context turns + current turn
9# Output: SPEAK or SILENT prediction for the target speaker1@misc{bhagtani2026speakstaysilentcontextaware,
2 title={Speak or Stay Silent: Context-Aware Turn-Taking in Multi-Party Dialogue},
3 author={Bhagtani, Kratika and Anand, Mrinal and Xu, Yu Chen and Yadav, Amit Kumar Singh},
4 year={2026},
5 archivePrefix={arXiv},
6 url={https://arxiv.org/abs/2603.11409}
7}