def METHOD_NAME(self, device, seed=0):
    if str(device).startswith("mps"):
        generator = torch.manual_seed(seed)
    else:
        generator = torch.Generator(device=device).manual_seed(seed)
    inputs = {
        "batch_size": 1,
        "generator": generator,
        "num_inference_steps": 4,
    }
    return inputs