Fix bug in make_preprocessor_external

#3
Files changed (1) hide show
  1. 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