alidev2002 commited on
Commit
8003e42
Β·
verified Β·
1 Parent(s): f2dcb28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -53,7 +53,7 @@ def test_stream(text):
53
 
54
  print("54")
55
 
56
- partial = token_ids[i].unsqueeze(0)
57
 
58
  print(f"58: {partial}")
59
 
 
53
 
54
  print("54")
55
 
56
+ partial = token_ids[i]
57
 
58
  print(f"58: {partial}")
59