Hugging Face's logo

Spaces:
tenet
/
rwkv-mcts-cot
Runtime error

App Files Community
tenet commited on
Commit
fbae8c9
•
1 Parent(s): c0828ba

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ import random
4
  import numpy as np
5
  import gradio as gr
6
  from transformers import AutoTokenizer
7
- from rwkv import RWKV
8
 
9
  # Define the Node class for MCTS
10
  class Node:
 
4
  import numpy as np
5
  import gradio as gr
6
  from transformers import AutoTokenizer
7
+ from rwkv.model import RWKV
8
 
9
  # Define the Node class for MCTS
10
  class Node: