Views
No views yet
1import torch
2from examples.mac_m3_grpo import WorldModel, PolicyNetwork
3
4# Initialize the model
5world_model = WorldModel(
6 vocab_size=102,
7 embed_dim=64,
8 latent_dim=32
9)
10
11# Load the weights
12world_model.load_state_dict(torch.load("world_model.pt"))
13
14# Create policy
15policy = PolicyNetwork(world_model)
16
17# Generate text
18# [Your generation code here]