From 0f6708c471d3e3daa507067b214df8556b799a52 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 1 Mar 2023 12:18:17 -0300
Subject: [PATCH] Sort the imports

---
 modules/RWKV.py            | 1 +
 modules/text_generation.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/RWKV.py b/modules/RWKV.py
index 9f348ad7..c5424d8f 100644
--- a/modules/RWKV.py
+++ b/modules/RWKV.py
@@ -16,6 +16,7 @@ os.environ["RWKV_CUDA_ON"] = '0' #  '1' : use CUDA kernel for seq mode (much fas
 from rwkv.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 
+
 class RWKVModel:
     def __init__(self):
         pass
diff --git a/modules/text_generation.py b/modules/text_generation.py
index 1bd84cd2..4c9d1f0e 100644
--- a/modules/text_generation.py
+++ b/modules/text_generation.py
@@ -5,8 +5,8 @@ import time
 import numpy as np
 import torch
 import transformers
-from tqdm import tqdm
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
+from tqdm import tqdm
 
 import modules.shared as shared
 from modules.extensions import apply_extensions