From ff9f649c0c588ba91be9fc1c75e2d3266c81c474 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 2 Mar 2023 00:36:20 -0300
Subject: [PATCH] Remove some unused imports

---
 modules/RWKV.py            | 3 ---
 modules/text_generation.py | 1 -
 2 files changed, 4 deletions(-)

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