Files changed (1) hide show
  1. README.md +0 -1
README.md CHANGED
@@ -52,7 +52,6 @@ model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
52
  generated_ids = model.generate(
53
  **model_inputs,
54
  max_new_tokens=1024,
55
- temperature=0.6
56
  )
57
  generated_ids = [
58
  output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
 
52
  generated_ids = model.generate(
53
  **model_inputs,
54
  max_new_tokens=1024,
 
55
  )
56
  generated_ids = [
57
  output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)