Update quantizedln.md

kyegomez · Dec 27, 2023 · 35864c9 · 35864c9
1 parent eb9606a
commit 35864c9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/zeta/nn/modules/quantizedln.md b/docs/zeta/nn/modules/quantizedln.md
@@ -82,6 +82,8 @@ Define a custom network that uses have the `QuantizedLN` module:
 
 ```python
 import torch.nn as nn
+from zeta.nn.modules import QuantizedLN
+
 
 class CustomNetwork(nn.Module):
     def __init__(self):
@@ -110,6 +112,8 @@ The `QuantizedLN` module in a multi-layer setup:
 
 ```python
 import torch.nn as nn
+from zeta.nn.modules import QuantizedLN
+
 
 class DeepNetwork(nn.Module):
     def __init__(self):