Unverified Commit cb3e5c33 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix a few last paths for the new repo org (#8666)

parent a79a96dd
...@@ -1055,7 +1055,7 @@ If you need to validate the output of a logger, you can use :obj:`CaptureLogger` ...@@ -1055,7 +1055,7 @@ If you need to validate the output of a logger, you can use :obj:`CaptureLogger`
msg = "Testing 1, 2, 3" msg = "Testing 1, 2, 3"
logging.set_verbosity_info() logging.set_verbosity_info()
logger = logging.get_logger("transformers.tokenization_bart") logger = logging.get_logger("transformers.models.bart.tokenization_bart")
with CaptureLogger(logger) as cl: with CaptureLogger(logger) as cl:
logger.info(msg) logger.info(msg)
assert cl.out, msg+"\n" assert cl.out, msg+"\n"
......
...@@ -3,7 +3,7 @@ import json ...@@ -3,7 +3,7 @@ import json
from typing import List from typing import List
from ltp import LTP from ltp import LTP
from transformers.tokenization_bert import BertTokenizer from transformers import BertTokenizer
def _is_chinese_char(cp): def _is_chinese_char(cp):
......
...@@ -107,7 +107,12 @@ def make_support(question, source="wiki40b", method="dense", n_results=10): ...@@ -107,7 +107,12 @@ def make_support(question, source="wiki40b", method="dense", n_results=10):
return question_doc, support_list return question_doc, support_list
@st.cache(hash_funcs={torch.Tensor: (lambda _: None), transformers.tokenization_bart.BartTokenizer: (lambda _: None)}) @st.cache(
hash_funcs={
torch.Tensor: (lambda _: None),
transformers.models.bart.tokenization_bart.BartTokenizer: (lambda _: None),
}
)
def answer_question( def answer_question(
question_doc, s2s_model, s2s_tokenizer, min_len=64, max_len=256, sampling=False, n_beams=2, top_p=0.95, temp=0.8 question_doc, s2s_model, s2s_tokenizer, min_len=64, max_len=256, sampling=False, n_beams=2, top_p=0.95, temp=0.8
): ):
......
...@@ -9,8 +9,7 @@ tags: ...@@ -9,8 +9,7 @@ tags:
```python ```python
import json import json
import os import os
from transformers.configuration_roberta import RobertaConfig from transformers import RobertaConfig, RobertaForMaskedLM, TFRobertaForMaskedLM
from transformers import RobertaForMaskedLM, TFRobertaForMaskedLM
DIRNAME = "./dummy-unknown" DIRNAME = "./dummy-unknown"
......
...@@ -470,7 +470,7 @@ class CaptureLogger: ...@@ -470,7 +470,7 @@ class CaptureLogger:
>>> msg = "Testing 1, 2, 3" >>> msg = "Testing 1, 2, 3"
>>> logging.set_verbosity_info() >>> logging.set_verbosity_info()
>>> logger = logging.get_logger("transformers.tokenization_bart") >>> logger = logging.get_logger("transformers.models.bart.tokenization_bart")
>>> with CaptureLogger(logger) as cl: >>> with CaptureLogger(logger) as cl:
... logger.info(msg) ... logger.info(msg)
>>> assert cl.out, msg+"\n" >>> assert cl.out, msg+"\n"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment