RWKV docs

labmlai · Mar 17, 2024 · 418d1ec · 418d1ec
1 parent df9e1af
commit 418d1ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/labml_nn/RWKV/experiment.py b/labml_nn/RWKV/experiment.py
@@ -5,8 +5,8 @@
 import torch.nn as nn
 from labml_nn.RWKV.configs import RWKVConfigs
 
-from __init__ import RWKV
-from __init__ import TimeMixing
+from labml_nn.RWKV import RWKV
+from labml_nn.RWKV import TimeMixing
 from labml import experiment
 from labml.configs import option
 from labml_nn.experiments.nlp_autoregression import NLPAutoRegressionConfigs