Hugging Face's logo

Spaces:
3ZadeSSG
/
RT-MPINet
Sleeping

App Files Community
3ZadeSSG commited on
Commit
92df483
·
1 Parent(s): b9062e3

updated imports

Browse files
Files changed (2) hide show
  1. model_Large.py +0 -1
  2. model_Medium.py +0 -1
model_Large.py CHANGED
@@ -5,7 +5,6 @@ import warnings
5
  warnings.filterwarnings("ignore")
6
  import torchvision
7
  import parameters as params
8
- import timm
9
 
10
  class DinoV2FeatureExtractor(nn.Module):
11
  def __init__(self, out_channels=256, out_size=(64, 64)):
 
5
  warnings.filterwarnings("ignore")
6
  import torchvision
7
  import parameters as params
 
8
 
9
  class DinoV2FeatureExtractor(nn.Module):
10
  def __init__(self, out_channels=256, out_size=(64, 64)):
model_Medium.py CHANGED
@@ -5,7 +5,6 @@ import warnings
5
  warnings.filterwarnings("ignore")
6
  import torchvision
7
  import parameters as params
8
- import timm
9
 
10
  class DinoV2FeatureExtractor(nn.Module):
11
  def __init__(self, out_channels=256, out_size=(64, 64)):
 
5
  warnings.filterwarnings("ignore")
6
  import torchvision
7
  import parameters as params
 
8
 
9
  class DinoV2FeatureExtractor(nn.Module):
10
  def __init__(self, out_channels=256, out_size=(64, 64)):