Spaces:

pedrocas15
/

RPC-Chat

Runtime error

pedrocas15 commited on Feb 28

Commit

b2165fc

verified ·

1 Parent(s): 7d65953

Update rpc.py

Files changed (1) hide show

rpc.py CHANGED Viewed

@@ -181,7 +181,7 @@ def load_index(index_path="/dev/shm/rpc-vecdb/index"):
 # Generate Function
 def generate(text, use_rpc=True, max_tokens=128):
     enc_text = tokenizer.encode(text, add_special_tokens=False)
     i = 0
     while i < max_tokens and tok != vocab_size - 2:

 # Generate Function
 def generate(text, use_rpc=True, max_tokens=128):
     enc_text = tokenizer.encode(text, add_special_tokens=False)
+    tok = None
     i = 0
     while i < max_tokens and tok != vocab_size - 2: