From 418d1ec44a7abca051bb7f7e55ad593ba684433d Mon Sep 17 00:00:00 2001
From: Varuna Jayasiri <vpjayasiri@gmail.com>
Date: Sun, 17 Mar 2024 17:47:39 +0530
Subject: [PATCH] RWKV docs

---
 labml_nn/RWKV/experiment.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/labml_nn/RWKV/experiment.py b/labml_nn/RWKV/experiment.py
index ad18fe3b..1f99d66d 100644
--- 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