wanicca commited on
Commit
290c9c9
·
1 Parent(s): aa598ee

fix pynvml

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
  '''