Spaces:

houwei-chou
/

demo

Runtime error

joechou commited on Jan 25, 2023

Commit

b82831b

•

1 Parent(s): e588819

Update dataset.py

Files changed (1) hide show

dataset.py CHANGED Viewed

@@ -112,7 +112,7 @@ def from_processed(url: str, train=False):
 def get_collate_fn(text_vocab_size: int, max_seq_length: int):
     def collate_fn(examples: List[Example]) -> Batched:
-        from kobe.data.vocab import BOS_ID, EOS_ID
         title_token_ids = pad_sequence(
             [

 def get_collate_fn(text_vocab_size: int, max_seq_length: int):
     def collate_fn(examples: List[Example]) -> Batched:
+        from vocab import BOS_ID, EOS_ID
         title_token_ids = pad_sequence(
             [