gen.py

import numpy as np
import os

# Tensor shapes
shapes = {
    "arg0": (2816, 2),
    "arg1": (1280, 2816),
}

# Generate and save tensors
for name, shape in shapes.items():
    tensor = np.random.uniform(low=0.0, high=1.0, size=shape).astype(np.float16)
    np.save(f"{name}.npy", tensor)
    print(f"Saved {name}.npy with shape {shape}")