Vision-CAIR
commited on
Commit
•
ad32631
1
Parent(s):
4559ee4
Upload folder using huggingface_hub
Browse files- blip_processors.py +3 -3
- config.py +1 -1
- conversation.py +1 -1
- eva_vit.py +1 -1
- optims.py +1 -1
blip_processors.py
CHANGED
@@ -7,9 +7,9 @@
|
|
7 |
|
8 |
import re
|
9 |
|
10 |
-
from
|
11 |
-
from
|
12 |
-
from
|
13 |
from omegaconf import OmegaConf
|
14 |
from torchvision import transforms
|
15 |
from torchvision.transforms.functional import InterpolationMode
|
|
|
7 |
|
8 |
import re
|
9 |
|
10 |
+
from .registry import registry
|
11 |
+
from .base_processor import BaseProcessor
|
12 |
+
from .randaugment import RandomAugment
|
13 |
from omegaconf import OmegaConf
|
14 |
from torchvision import transforms
|
15 |
from torchvision.transforms.functional import InterpolationMode
|
config.py
CHANGED
@@ -10,7 +10,7 @@ import json
|
|
10 |
from typing import Dict
|
11 |
|
12 |
from omegaconf import OmegaConf
|
13 |
-
from
|
14 |
|
15 |
|
16 |
class Config:
|
|
|
10 |
from typing import Dict
|
11 |
|
12 |
from omegaconf import OmegaConf
|
13 |
+
from .registry import registry
|
14 |
|
15 |
|
16 |
class Config:
|
conversation.py
CHANGED
@@ -10,7 +10,7 @@ import dataclasses
|
|
10 |
from enum import auto, Enum
|
11 |
from typing import List, Tuple, Any
|
12 |
|
13 |
-
from
|
14 |
|
15 |
|
16 |
class SeparatorStyle(Enum):
|
|
|
10 |
from enum import auto, Enum
|
11 |
from typing import List, Tuple, Any
|
12 |
|
13 |
+
from .registry import registry
|
14 |
|
15 |
|
16 |
class SeparatorStyle(Enum):
|
eva_vit.py
CHANGED
@@ -15,7 +15,7 @@ import torch.utils.checkpoint as checkpoint
|
|
15 |
from timm.models.layers import drop_path, to_2tuple, trunc_normal_
|
16 |
from timm.models.registry import register_model
|
17 |
|
18 |
-
from
|
19 |
|
20 |
def _cfg(url='', **kwargs):
|
21 |
return {
|
|
|
15 |
from timm.models.layers import drop_path, to_2tuple, trunc_normal_
|
16 |
from timm.models.registry import register_model
|
17 |
|
18 |
+
from .dist_utils import download_cached_file
|
19 |
|
20 |
def _cfg(url='', **kwargs):
|
21 |
return {
|
optims.py
CHANGED
@@ -7,7 +7,7 @@
|
|
7 |
|
8 |
import math
|
9 |
|
10 |
-
from
|
11 |
|
12 |
|
13 |
@registry.register_lr_scheduler("linear_warmup_step_lr")
|
|
|
7 |
|
8 |
import math
|
9 |
|
10 |
+
from .registry import registry
|
11 |
|
12 |
|
13 |
@registry.register_lr_scheduler("linear_warmup_step_lr")
|