fix pynvml
Browse files
app.py
CHANGED
@@ -3,9 +3,9 @@ import argparse
|
|
3 |
import os, gc, torch
|
4 |
from datetime import datetime
|
5 |
from huggingface_hub import hf_hub_download
|
6 |
-
from pynvml import *
|
7 |
-
nvmlInit()
|
8 |
-
gpu_h = nvmlDeviceGetHandleByIndex(0)
|
9 |
ctx_limit = 4096
|
10 |
desc = f'''链接:<a href='https://github.com/BlinkDL/ChatRWKV' target="_blank" style="margin:0 0.5em">ChatRWKV</a><a href='https://github.com/BlinkDL/RWKV-LM' target="_blank" style="margin:0 0.5em">RWKV-LM</a><a href="https://pypi.org/project/rwkv/" target="_blank" style="margin:0 0.5em">RWKV pip package</a><a href="https://zhuanlan.zhihu.com/p/618011122" target="_blank" style="margin:0 0.5em">知乎教程</a>
|
11 |
'''
|
|
|
3 |
import os, gc, torch
|
4 |
from datetime import datetime
|
5 |
from huggingface_hub import hf_hub_download
|
6 |
+
# from pynvml import *
|
7 |
+
# nvmlInit()
|
8 |
+
# gpu_h = nvmlDeviceGetHandleByIndex(0)
|
9 |
ctx_limit = 4096
|
10 |
desc = f'''链接:<a href='https://github.com/BlinkDL/ChatRWKV' target="_blank" style="margin:0 0.5em">ChatRWKV</a><a href='https://github.com/BlinkDL/RWKV-LM' target="_blank" style="margin:0 0.5em">RWKV-LM</a><a href="https://pypi.org/project/rwkv/" target="_blank" style="margin:0 0.5em">RWKV pip package</a><a href="https://zhuanlan.zhihu.com/p/618011122" target="_blank" style="margin:0 0.5em">知乎教程</a>
|
11 |
'''
|