I want a generate(model: torch.nn.Module, tokenizer: Any, input text: str, max length: int, temperature…
I want a generate(model: torch.nn.Module, tokenizer: Any, input text: str, max length: int, temperature…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). I want a generate(model: torch.nn.Module, tokenizer: Any, input text: str, max length: int, temperature: float = 1.2, top…
The task
I want a generate(model: torch.nn.Module, tokenizer: Any, input_text: str, max_length: int, temperature: float = 1.2, top_p: float = 0.95, top_k: int = 0, scaler_state: Optional[Dict[str, Any]] = None, device: torch.device = torch.device("mps")) -> str function for…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.