Spaces:

flax-community
/

dalle-mini

Running

boris commited on Jul 15, 2021

Commit

ca83cca

•

1 Parent(s): 3d61350

fix: typo

Files changed (1) hide show

seq2seq/run_seq2seq_flax.py CHANGED Viewed

@@ -435,7 +435,7 @@ def main():
     )
     if model_args.from_checkpoint is not None:
-        artifact = run.use_artifact('wandb/hf-flax-dalle-mini/model-3h3x3565:latest')
         artifact_dir = artifact.download()
         model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)

     )
     if model_args.from_checkpoint is not None:
+        artifact = wandb.run.use_artifact('wandb/hf-flax-dalle-mini/model-3h3x3565:latest')
         artifact_dir = artifact.download()
         model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)