boris commited on
Commit
e803feb
1 Parent(s): ca83cca

fix: custom reference

Browse files
Files changed (1) hide show
  1. seq2seq/run_seq2seq_flax.py +1 -1
seq2seq/run_seq2seq_flax.py CHANGED
@@ -435,7 +435,7 @@ def main():
435
  )
436
 
437
  if model_args.from_checkpoint is not None:
438
- artifact = wandb.run.use_artifact('wandb/hf-flax-dalle-mini/model-3h3x3565:latest')
439
  artifact_dir = artifact.download()
440
  model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)
441
 
 
435
  )
436
 
437
  if model_args.from_checkpoint is not None:
438
+ artifact = wandb.run.use_artifact(model_args.from_checkpoint)
439
  artifact_dir = artifact.download()
440
  model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)
441