Sort the imports
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user