code: def test_forward(self, model, tokens, use_pf, fp16):
      output = model(**tokens, output_hidden_states=True)
      print("forward:", output.hidden_states[0][0])
      torch.save(
          output.hidden_states,
          f"output-{'pf' if use_pf else 'non-pf'}-{'fp16' if fp16 else 'fp32'}.pt",
      )
      assert isinstance(output.hidden_states[0][0], torch.Tensor)
  
file_path: parallelformers/tests/ibert/ibert_model.py
imports: import os
  import unittest
  from argparse import ArgumentParser
  import torch
  from transformers import IBertModel, RobertaTokenizer
  from parallelformers import parallelize
label_desc: Use image node_id 1 for calling the TestModel obj's underlying member method code with example usage: obj.test_forward(model, tokens, use_pf, fp16) without return types