update Adapter class

adapter-hub · Oct 23, 2024 · ca9905b · ca9905b
1 parent 0660776
commit ca9905b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/adapters/methods/modeling.py b/src/adapters/methods/modeling.py
@@ -7,7 +7,7 @@
 
 from ..configuration import AdapterFusionConfig, BnConfig
 from ..context import ForwardContext
-from vision import StochasticDepth
+from .vision import StochasticDepth
 
 
 class Activation_Function_Class(nn.Module):
@@ -144,8 +144,8 @@ def __init__(
         else:
             raise ValueError("Unknown init_weights type: {}".format(config["init_weights"]))
 
-        if config["drop_path"] > 0.0:
-            self.DropPath = StochasticDepth(drop_prob=config["drop_path"])
+        if config["stochastic_depth"] > 0.0:
+            self.DropPath = StochasticDepth(drop_prob=config["stochastic_depth"])
 
     def pre_forward(
         self,