sagawa
/

ReactionT5v2-yield

Model card Files Files and versions Community

sagawa commited on 14 days ago

Commit

802c202

•

1 Parent(s): 38930a7

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -36,6 +36,8 @@ Use the code below to get started with the model.
 import torch
 import torch.nn as nn
 from transformers import AutoTokenizer, T5ForConditionalGeneration, AutoConfig, PreTrainedModel
 class ReactionT5Yield(PreTrainedModel):
     config_class  = AutoConfig

 import torch
 import torch.nn as nn
 from transformers import AutoTokenizer, T5ForConditionalGeneration, AutoConfig, PreTrainedModel
+import logging
+logging.getLogger('transformers').setLevel(logging.ERROR)
 class ReactionT5Yield(PreTrainedModel):
     config_class  = AutoConfig