Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files- app_new.py +2 -1
- 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)
|