KingNish commited on
Commit
9a02bb8
·
verified ·
1 Parent(s): d159d0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -27,10 +27,11 @@ from modeling.qwen2 import Qwen2Tokenizer
27
 
28
  from huggingface_hub import snapshot_download
29
 
30
- save_dir = "/model"
31
  repo_id = "ByteDance-Seed/BAGEL-7B-MoT"
 
32
 
33
- snapshot_download(
34
  local_dir=save_dir,
35
  repo_id=repo_id,
36
  local_dir_use_symlinks=False,
 
27
 
28
  from huggingface_hub import snapshot_download
29
 
30
+ save_dir = "./model"
31
  repo_id = "ByteDance-Seed/BAGEL-7B-MoT"
32
+ cache_dir = save_dir + "/cache"
33
 
34
+ snapshot_download(cache_dir=cache_dir,
35
  local_dir=save_dir,
36
  repo_id=repo_id,
37
  local_dir_use_symlinks=False,