A fix
Browse files
README.md
CHANGED
@@ -108,11 +108,9 @@ with torch.no_grad():
|
|
108 |
predicted_ids = torch.argmax(logits, dim=-1)
|
109 |
predictions = processor.batch_decode(predicted_ids)
|
110 |
|
111 |
-
# Log
|
112 |
-
print('---')
|
113 |
print('Predictions:')
|
114 |
print(predictions)
|
115 |
-
print('---')
|
116 |
```
|
117 |
|
118 |
### Licenses
|
|
|
108 |
predicted_ids = torch.argmax(logits, dim=-1)
|
109 |
predictions = processor.batch_decode(predicted_ids)
|
110 |
|
111 |
+
# Log results
|
|
|
112 |
print('Predictions:')
|
113 |
print(predictions)
|
|
|
114 |
```
|
115 |
|
116 |
### Licenses
|