Fix bug in make_preprocessor_external
#3
by
mranzinger
- opened
- hf_model.py +1 -0
hf_model.py
CHANGED
@@ -16,6 +16,7 @@ from typing import Callable, Optional, List, Union
|
|
16 |
|
17 |
from timm.models import VisionTransformer
|
18 |
import torch
|
|
|
19 |
from transformers import PretrainedConfig, PreTrainedModel
|
20 |
|
21 |
|
|
|
16 |
|
17 |
from timm.models import VisionTransformer
|
18 |
import torch
|
19 |
+
from torch import nn
|
20 |
from transformers import PretrainedConfig, PreTrainedModel
|
21 |
|
22 |
|