Drexubery commited on
Commit
7cbba90
1 Parent(s): f80de23
Files changed (2) hide show
  1. app_new.py +2 -1
  2. configs/infer_config.py +1 -1
app_new.py CHANGED
@@ -10,7 +10,8 @@ from dust3r.inference import inference, load_model
10
  from omegaconf import OmegaConf
11
  from pytorch_lightning import seed_everything
12
  from utils.diffusion_utils import instantiate_from_config,load_model_checkpoint,image_guided_synthesis
13
-
 
14
 
15
  i2v_examples = [
16
  ['test/images/boy.png', 0, 1.0, '0 40', '0 0', '0 0', 50, 123],
 
10
  from omegaconf import OmegaConf
11
  from pytorch_lightning import seed_everything
12
  from utils.diffusion_utils import instantiate_from_config,load_model_checkpoint,image_guided_synthesis
13
+ import torchvision.transforms as transforms
14
+ import copy
15
 
16
  i2v_examples = [
17
  ['test/images/boy.png', 0, 1.0, '0 40', '0 0', '0 0', 50, 123],
configs/infer_config.py CHANGED
@@ -48,7 +48,7 @@ def get_parser():
48
  parser.add_argument("--n_samples", type=int, default=1, help="num of samples per prompt")
49
 
50
  ## dust3r
51
- parser.add_argument('--model_path', type=str, default='./DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth', help='The path of the model')
52
  parser.add_argument('--batch_size', default=1)
53
  parser.add_argument('--schedule', type=str, default='linear')
54
  parser.add_argument('--niter', default=300)
 
48
  parser.add_argument("--n_samples", type=int, default=1, help="num of samples per prompt")
49
 
50
  ## dust3r
51
+ parser.add_argument('--model_path', type=str, default='./checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth', help='The path of the model')
52
  parser.add_argument('--batch_size', default=1)
53
  parser.add_argument('--schedule', type=str, default='linear')
54
  parser.add_argument('--niter', default=300)