Update README.md
Browse files
README.md
CHANGED
|
@@ -36,6 +36,8 @@ Use the code below to get started with the model.
|
|
| 36 |
import torch
|
| 37 |
import torch.nn as nn
|
| 38 |
from transformers import AutoTokenizer, T5ForConditionalGeneration, AutoConfig, PreTrainedModel
|
|
|
|
|
|
|
| 39 |
|
| 40 |
class ReactionT5Yield(PreTrainedModel):
|
| 41 |
config_class = AutoConfig
|
|
|
|
| 36 |
import torch
|
| 37 |
import torch.nn as nn
|
| 38 |
from transformers import AutoTokenizer, T5ForConditionalGeneration, AutoConfig, PreTrainedModel
|
| 39 |
+
import logging
|
| 40 |
+
logging.getLogger('transformers').setLevel(logging.ERROR)
|
| 41 |
|
| 42 |
class ReactionT5Yield(PreTrainedModel):
|
| 43 |
config_class = AutoConfig
|