Updated README

README.md

@@ -85,7 +85,6 @@ prompt = f'''What is the difference between nuclear fusion and fission?
 ###Response:'''
 
 input_ids = tokenizer(prompt, return_tensors='pt').input_ids.cuda()
-t1 = time.time()
 output = model.generate(
     inputs=input_ids, 
     temperature=0.7,
@@ -95,9 +94,6 @@ output = model.generate(
     repetition_penalty=1.1,
     eos_token_id=tokenizer.eos_token_id,
     streamer=streamer)
-t2 = time.time()
-print("*"*80)
-print(f"Generated {num_tokens/(t2-t1):.2f} token/s; {(t2-t1)*1000/num_tokens:.2f} ms/token")
 ```
 
 ## Evaluation