Method save

lit_gpt/utils.py:243–252

(self, obj)

        raise TypeError(f'can only store tensors early, not {type(tensor)}')

    def save(self, obj):
        if self.has_saved:
            raise RuntimeError('have already saved')
        # Write the pickle data for `obj`
        data_buf = BytesIO()
        pickler = IncrementalPyTorchPickler(self, data_buf, protocol=5)
        pickler.dump(obj)
        data_value = data_buf.getvalue()
        self.zipfile.write_record('data.pkl', data_value, len(data_value))
        self.has_saved = True

    def _write_storage_and_return_key(self, storage):
        if self.has_saved: