Spaces:
Running
on
Zero
Running
on
Zero
Change to IN3D trained adapter.
Browse files
app.py
CHANGED
@@ -182,7 +182,7 @@ def reload_img(
|
|
182 |
# βββ Configuration βββββββββββββββββββββββββββββββββββββββββββββββ
|
183 |
num_patches = 45
|
184 |
target_res = num_patches * 14
|
185 |
-
ckpt_file = "./ckpts/
|
186 |
|
187 |
# βββ Model setup βββββββββββββββββββββββββββββββββββββββββββββββββ
|
188 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
182 |
# βββ Configuration βββββββββββββββββββββββββββββββββββββββββββββββ
|
183 |
num_patches = 45
|
184 |
target_res = num_patches * 14
|
185 |
+
ckpt_file = "./ckpts/dino_in3d_0712.pth" #dino_spair_0300
|
186 |
|
187 |
# βββ Model setup βββββββββββββββββββββββββββββββββββββββββββββββββ
|
188 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|