Hugging Face's logo

Spaces:
csgaobb
/
MetaUAS
Running

App Files Community
1
csgaobb commited on
Commit
618fc8b
·
1 Parent(s): 549bb2c

fix bug

Browse files
Files changed (2) hide show
  1. metauas.py +1 -1
  2. requirements.txt +3 -1
metauas.py CHANGED
@@ -179,7 +179,7 @@ class MetaUAS(pl.LightningModule):
179
  decoder_channels=decoder_channels,
180
  n_blocks= self.decoder_depth,
181
  attention_type="scse",
182
- #num_coam_layers= num_alignment_layers,
183
  )
184
 
185
  elif self.decoder_name == "fpn":
 
179
  decoder_channels=decoder_channels,
180
  n_blocks= self.decoder_depth,
181
  attention_type="scse",
182
+ num_coam_layers= num_alignment_layers,
183
  )
184
 
185
  elif self.decoder_name == "fpn":
requirements.txt CHANGED
@@ -8,7 +8,9 @@ opencv_python==4.6.0.66
8
  opencv_python_headless==4.7.0.72
9
  Pillow==8.4.0
10
  pytorch_lightning==1.9.0
11
- segmentation_models_pytorch==0.2.1
12
  torch==1.12.1
13
  torchvision==0.13.1
14
  tqdm==4.62.3
 
 
 
8
  opencv_python_headless==4.7.0.72
9
  Pillow==8.4.0
10
  pytorch_lightning==1.9.0
11
+ #segmentation_models_pytorch==0.2.1
12
  torch==1.12.1
13
  torchvision==0.13.1
14
  tqdm==4.62.3
15
+ segmentation-models-pytorch@git+https://github.com/ragavsachdeva/segmentation_models.pytorch.git@0092ee4d6f851d89a4a401bb2dfa6187660b8dd3
16
+