From 215629ced63a5601bbcdc434a135608c8ff45ffe Mon Sep 17 00:00:00 2001 From: Zhiqi Bu Date: Mon, 19 Feb 2024 13:14:47 +0800 Subject: [PATCH] Update privacy_engine_dist_extending.py --- fastDP/privacy_engine_dist_extending.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/fastDP/privacy_engine_dist_extending.py b/fastDP/privacy_engine_dist_extending.py index 3b112d5..a4f5c7d 100644 --- a/fastDP/privacy_engine_dist_extending.py +++ b/fastDP/privacy_engine_dist_extending.py @@ -16,14 +16,12 @@ from .accounting import accounting_manager from torch.functional import F import transformers -import mstar.models.gpt2_model from .supported_differentially_private_layers import * _DP_supported_layers = [nn.Embedding, nn.Linear, nn.Conv2d, nn.LayerNorm, nn.GroupNorm, transformers.pytorch_utils.Conv1D, - mstar.models.gpt2_model.Conv1DGeLU, ] class PrivacyEngine_Distributed_extending(object):