Spaces:

project-baize
/

chat-with-baize

Runtime error

project-baize commited on Apr 4, 2023

Commit

b860d5b

•

1 Parent(s): 9d3530f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,6 +44,7 @@ def predict(text,
         begin_length = len(prompt)
     torch.cuda.empty_cache()
     input_ids = inputs["input_ids"].to(device)
     total_count += 1
     print(total_count)
     with torch.no_grad():

         begin_length = len(prompt)
     torch.cuda.empty_cache()
     input_ids = inputs["input_ids"].to(device)
+    global total_count
     total_count += 1
     print(total_count)
     with torch.no_grad():