ydshieh HF staff commited on
Commit
679d099
1 Parent(s): 8f85ccf

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -26,7 +26,7 @@ files_to_download = [
26
 
27
  # copy files from checkpoint hub:
28
  for fn in files_to_download:
29
- file_path = hf_hub_download("ydshieh/vit-gpt2-coco-en", f"ckpt_epoch_3_step_6900/{fn}")
30
  shutil.copyfile(file_path, os.path.join(model_dir, fn))
31
 
32
  model = FlaxVisionEncoderDecoderModel.from_pretrained(model_dir)
 
26
 
27
  # copy files from checkpoint hub:
28
  for fn in files_to_download:
29
+ file_path = hf_hub_download("ydshieh/vit-gpt2-coco-en-ckpts", f"ckpt_epoch_3_step_6900/{fn}")
30
  shutil.copyfile(file_path, os.path.join(model_dir, fn))
31
 
32
  model = FlaxVisionEncoderDecoderModel.from_pretrained(model_dir)