class_docstr: Test `FeaturesManager.determine_framework`
class_docstr_tok: ["Test","`","FeaturesManager.determine_framework","`"]
code: def _setup_pt_ckpt(self, save_dir):
      model_pt = AutoModel.from_pretrained(self.test_model)
      model_pt.save_pretrained(save_dir)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/tests/onnx/test_features.py
imports: from tempfile import TemporaryDirectory
  from unittest import TestCase
  from unittest.mock import MagicMock, patch
  from transformers import AutoModel, TFAutoModel
  from transformers.onnx import FeaturesManager
  from transformers.testing_utils import SMALL_MODEL_IDENTIFIER, require_tf, require_torch
label_desc: Use image node_id 2 for calling the DetermineFrameworkTest obj's underlying member method code with example usage: obj._setup_pt_ckpt(save_dir) without return types