John Ho commited on
Commit
4dc5aed
·
1 Parent(s): c5ab6e1

debugging internvl3

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -214,6 +214,7 @@ def inference(
214
  ).to("cuda", dtype=DTYPE)
215
 
216
  output = model.generate(**inputs, max_new_tokens=max_tokens)
 
217
  output_text = processor.decode(
218
  output[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True
219
  )
 
214
  ).to("cuda", dtype=DTYPE)
215
 
216
  output = model.generate(**inputs, max_new_tokens=max_tokens)
217
+ logger.debug(output)
218
  output_text = processor.decode(
219
  output[0, inputs["input_ids"].shape[1] :], skip_special_tokens=True
220
  )