Update privacy_engine_dist_extending.py

awslabs · Feb 19, 2024 · 215629c · 215629c
1 parent b4068af
commit 215629c
Showing 1 changed file with 0 additions and 2 deletions.
diff --git 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):