Method test_save_load_model

tests/others/test_flashpack.py:34–39

(self)

    @require_flashpack
    def test_save_load_model(self):
        model = AutoModel.from_pretrained(self.model_id, subfolder="transformer")
        with tempfile.TemporaryDirectory() as temp_dir:
            model.save_pretrained(temp_dir, use_flashpack=True)
            self.assertTrue((pathlib.Path(temp_dir) / "model.flashpack").exists())
            model = AutoModel.from_pretrained(temp_dir, use_flashpack=True)

    @require_flashpack
    def test_save_load_pipeline(self):