kjozsa commited on
Commit
beb6978
1 Parent(s): ca6382b

fix logging

Browse files
Files changed (1) hide show
  1. transformerschat.py +2 -2
transformerschat.py CHANGED
@@ -29,9 +29,9 @@ def ask(_, system_prompt, pre_prompt, question):
29
  'content': f"{question}",
30
  },
31
  ]
32
- logger.debug(f"<< {openhermes} << {question}")
33
  inputs = tokenizer(question, return_tensors="pt", return_attention_mask=False)
34
  outputs = model.generate(**inputs, max_length=200)
35
  answer = tokenizer.batch_decode(outputs)[0]
36
- logger.debug(f">> {openhermes} >> {answer}")
37
  return answer
 
29
  'content': f"{question}",
30
  },
31
  ]
32
+ logger.debug(f"<< openhermes << {question}")
33
  inputs = tokenizer(question, return_tensors="pt", return_attention_mask=False)
34
  outputs = model.generate(**inputs, max_length=200)
35
  answer = tokenizer.batch_decode(outputs)[0]
36
+ logger.debug(f">> openhermes >> {answer}")
37
  return answer