BehR-WorldModel-Textworld-Qwen2.5-7B

A behavior-consistent text-based world model for TextWorld.

Model details

Field Value
Base model X1AOX1A/WorldModel-Textworld-Qwen2.5-7B
Base revision b052a201ae867c3058efba17c9af9cb1635d1f09
Architecture Qwen2ForCausalLM (Qwen2.5-7B)
Environment TextWorld, via AgentGym

Given the agent-visible interaction history and the agent's next action, the model predicts the next environment observation. It can therefore act as a text simulator for agent rollouts in TextWorld.

Transformers usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "Ricardo-H/BehR-WorldModel-Textworld-Qwen2.5-7B"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    torch_dtype="auto",
    device_map="auto",
)

Evaluation

The paper evaluates this model on TextWorld with single-step exact match, task-success and consistency metrics. See the paper and the serving and evaluation repository for the evaluation protocol and reported results.

Intended use and limitations

  • Intended for research on world models and agent simulation in text-based environments.
  • Evaluated only on TextWorld; behavior outside that distribution is unvalidated.
  • Predicted observations may be plausible but factually wrong. Do not treat outputs as ground truth about the real environment.
  • The model inherits biases and failure modes from Qwen2.5-7B and the base world model.

License and provenance

This model is a derivative of X1AOX1A/WorldModel-Textworld-Qwen2.5-7B at revision b052a201ae867c3058efba17c9af9cb1635d1f09, which is itself derived from Qwen2.5-7B. Use is governed by the Qwen2.5 license and by the terms of the base world model; check the base repository before redistribution or commercial use. The serving and evaluation repository is Apache-2.0; these model weights are not.

The base world model and the WebShop/TextWorld data splits originate from From Word to World: Can Large Language Models be Implicit Text-based World Models? (arXiv:2512.18832).

Citation

@article{huang2026behrwm,
  title   = {Beyond State Consistency: Behavior Consistency in Text-Based World Models},
  author  = {Huang, Youling and Chen, Guanqiao and Yao, Junchi and Wang, Lu and
             Yang, Fangkai and Du, Chao and Zhao, ChenZhuo and Zhao, Pu and
             Lin, Qingwei and Rajmohan, Saravan and Zhang, Dongmei},
  journal = {arXiv preprint arXiv:2604.13824},
  year    = {2026},
  url     = {https://arxiv.org/abs/2604.13824}
}
Downloads last month
331
Safetensors
Model size
8B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for Ricardo-H/BehR-WorldModel-Textworld-Qwen2.5-7B

Base model

Qwen/Qwen2.5-7B
Finetuned
(1)
this model

Collection including Ricardo-H/BehR-WorldModel-Textworld-Qwen2.5-7B

Papers for Ricardo-H/BehR-WorldModel-Textworld-Qwen2.5-7B