JustinLin610 commited on
Commit
4e33a48
1 Parent(s): f1f58e7
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,13 +2,15 @@ import os
2
 
3
 
4
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
 
5
  'pip install --use-feature=in-tree-build ./; cd ..')
 
6
 
7
 
8
  import torch
9
  import numpy as np
10
  from fairseq import utils, tasks
11
- from fairseq.utils import checkpoint_utils
12
  from utils.eval_utils import eval_step
13
  from tasks.mm_tasks.caption import CaptionTask
14
  from models.ofa import OFAModel
 
2
 
3
 
4
  os.system('git clone https://github.com/pytorch/fairseq.git; cd fairseq;'
5
+ 'git checkout tags/v0.10.1;'
6
  'pip install --use-feature=in-tree-build ./; cd ..')
7
+ os.system('ls -l')
8
 
9
 
10
  import torch
11
  import numpy as np
12
  from fairseq import utils, tasks
13
+ from utils import checkpoint_utils
14
  from utils.eval_utils import eval_step
15
  from tasks.mm_tasks.caption import CaptionTask
16
  from models.ofa import OFAModel