[M1 Mac, Ventura 13.3.1, Python 3.10.4]
ChatRWKVのGUIアプリ製作に着手しました。
とりあえずPyQt6でガワだけ作成しました。chat.pyは改変してクラス化しています。


import os, copy, types, gc, sys
import numpy as np
from prompt_toolkit import prompt
import torch
class chat_base():
def __init__(self, lang, strat, model):
self.lang = lang
self.strat = strat
self.model = model
current_path = os.path.dirname(os.path.abspath(__file__))
sys.path.append(f'{current_path}/../rwkv_pip_package/src')
try:
os.environ["CUDA_VISIBLE_DEVICES"] = sys.argv[1]
except:
pass
np.set_printoptions(precision=4, suppress=True, linewidth=200)
args = types.SimpleNamespace()
print('\n\nChatRWKV v2 https://github.com/BlinkDL/ChatRWKV')
<以下略>