Update README.md
Browse files
README.md
CHANGED
|
@@ -62,7 +62,6 @@ with torch.no_grad():
|
|
| 62 |
output = merged_model.generate(**inputs, generation_config=generation_config)
|
| 63 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
| 64 |
|
| 65 |
-
print(f"Inference time: {end-start:.2f} seconds")
|
| 66 |
print(response)
|
| 67 |
```
|
| 68 |
|
|
|
|
| 62 |
output = merged_model.generate(**inputs, generation_config=generation_config)
|
| 63 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
| 64 |
|
|
|
|
| 65 |
print(response)
|
| 66 |
```
|
| 67 |
|