alexkueck commited on
Commit
8f4b344
·
1 Parent(s): d90f136

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -6
utils.py CHANGED
@@ -177,12 +177,7 @@ def predict(model,
177
  x = x.strip()
178
  a, b= [[y[0],convert_to_markdown(y[1])] for y in history]+[[text, convert_to_markdown(x)]],history + [[text,x]]
179
  antwort = antwort + a
180
- if shared_state.interrupted:
181
- shared_state.recover()
182
- try:
183
- return antwort
184
- except:
185
- pass
186
  del input_ids
187
  gc.collect()
188
  torch.cuda.empty_cache()
 
177
  x = x.strip()
178
  a, b= [[y[0],convert_to_markdown(y[1])] for y in history]+[[text, convert_to_markdown(x)]],history + [[text,x]]
179
  antwort = antwort + a
180
+
 
 
 
 
 
181
  del input_ids
182
  gc.collect()
183
  torch.cuda.empty_cache()