qianmuuq commited on
Commit
967f52d
1 Parent(s): fae408a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -153,7 +153,7 @@ def getChat(text,model,tokenizer):
153
  split_w = '×'
154
  else:
155
  split_w = '*'
156
- tt = s.split(split_w)
157
  f = 0
158
  if tt[0][0].isdigit():
159
  size_float = [float(i) for i in tt]
 
153
  split_w = '×'
154
  else:
155
  split_w = '*'
156
+ tt = size_str.split(split_w)
157
  f = 0
158
  if tt[0][0].isdigit():
159
  size_float = [float(i) for i in tt]