iioSnail commited on
Commit
5dea5bb
·
verified ·
1 Parent(s): b326139
Files changed (1) hide show
  1. modeling_glycebert.py +1 -3
modeling_glycebert.py CHANGED
@@ -19,7 +19,6 @@ from typing import List
19
  import numpy as np
20
  import torch
21
  from huggingface_hub import hf_hub_download
22
- from huggingface_hub.file_download import http_user_agent
23
  from torch import nn
24
  from torch.nn import CrossEntropyLoss, MSELoss
25
  from torch.nn import functional as F
@@ -47,8 +46,7 @@ def download_file(filename: str, path: Path):
47
  hf_hub_download(
48
  "iioSnail/ChineseBERT-base",
49
  filename,
50
- local_dir=cache_path,
51
- user_agent=http_user_agent(),
52
  )
53
  time.sleep(0.2)
54
 
 
19
  import numpy as np
20
  import torch
21
  from huggingface_hub import hf_hub_download
 
22
  from torch import nn
23
  from torch.nn import CrossEntropyLoss, MSELoss
24
  from torch.nn import functional as F
 
46
  hf_hub_download(
47
  "iioSnail/ChineseBERT-base",
48
  filename,
49
+ local_dir=cache_path
 
50
  )
51
  time.sleep(0.2)
52