Change import of MAE VIT
#10
by
eliasbela
- opened
- Prithvi_run_inference.py +1 -2
Prithvi_run_inference.py
CHANGED
|
@@ -9,8 +9,7 @@ import torch
|
|
| 9 |
import yaml
|
| 10 |
from einops import rearrange
|
| 11 |
|
| 12 |
-
from
|
| 13 |
-
|
| 14 |
|
| 15 |
NO_DATA = -9999
|
| 16 |
NO_DATA_FLOAT = 0.0001
|
|
|
|
| 9 |
import yaml
|
| 10 |
from einops import rearrange
|
| 11 |
|
| 12 |
+
from Prithvi import MaskedAutoencoderViT
|
|
|
|
| 13 |
|
| 14 |
NO_DATA = -9999
|
| 15 |
NO_DATA_FLOAT = 0.0001
|