diff --git a/modules/RWKV.py b/modules/RWKV.py
index aa4a0b9..46d8ff5 100644
--- a/modules/RWKV.py
+++ b/modules/RWKV.py
@@ -1,10 +1,7 @@
 import os
-import time
-import types
 from pathlib import Path
 
 import numpy as np
-import torch
 
 import modules.shared as shared
 
diff --git a/modules/text_generation.py b/modules/text_generation.py
index cdf4adf..3aabca9 100644
--- a/modules/text_generation.py
+++ b/modules/text_generation.py
@@ -5,7 +5,6 @@ import time
 import numpy as np
 import torch
 import transformers
-from rwkv.utils import PIPELINE, PIPELINE_ARGS
 from tqdm import tqdm
 
 import modules.shared as shared