prhegde
/

t5-query-reformulation-RL

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

prhegde commited on Apr 24

Commit

cee8c26

•

1 Parent(s): 5b8cf96

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -15,6 +15,8 @@ inference_config:
     num_beams: 1
     do_sample: true
     repetition_penalty: 1.8
 ---
 ## Model Summary
@@ -62,4 +64,4 @@ with torch.no_grad():
         output = model.generate(input_ids, max_length=35, num_beams=1, do_sample=True, repetition_penalty=1.8)
         target_sequence = tokenizer.decode(output[0], skip_special_tokens=True)
         print(f'Target: {target_sequence}')
-```

     num_beams: 1
     do_sample: true
     repetition_penalty: 1.8
+tags:
+- code
 ---
 ## Model Summary
         output = model.generate(input_ids, max_length=35, num_beams=1, do_sample=True, repetition_penalty=1.8)
         target_sequence = tokenizer.decode(output[0], skip_special_tokens=True)
         print(f'Target: {target_sequence}')
+```