Update slerpmodelmerger.md

kyegomez · Dec 27, 2023 · e54e15d · e54e15d
1 parent f3f3de9
commit e54e15d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/zeta/nn/modules/slerpmodelmerger.md b/docs/zeta/nn/modules/slerpmodelmerger.md
@@ -42,6 +42,8 @@ The following code shows how to use the SLERPModelMerger class to merge two PyTo
 
 ```python
 import torch.nn as nn
+from zeta.nn import SLERPModelMerger
+
 model1 = nn.Linear(10, 10)
 model2 = nn.Linear(10, 10)