@@ -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 tqdm import tqdm
import modules.shared as shared
from modules.extensions import apply_extensions