diff options
| author | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
|---|---|---|
| committer | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
| commit | 6ebe04327397ebf7c9400d68c43fede705f8ce75 (patch) | |
| tree | 4e701ef55370cb6226593019f9e6daaf8ddf73b4 /candle-transformers/src/models/segment_anything/tiny_vit.rs | |
| parent | 6bf52b9fdf82ad775611e82924d73172660a605e (diff) | |
| parent | 84250bf52f58528cf59dca3b82effd9f07a13cc7 (diff) | |
| download | candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.gz candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.bz2 candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.zip | |
Merge branch 'main' into ivarflakstad/metal-prng
Diffstat (limited to 'candle-transformers/src/models/segment_anything/tiny_vit.rs')
| -rw-r--r-- | candle-transformers/src/models/segment_anything/tiny_vit.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-transformers/src/models/segment_anything/tiny_vit.rs b/candle-transformers/src/models/segment_anything/tiny_vit.rs index cd2936ab..d1700cc5 100644 --- a/candle-transformers/src/models/segment_anything/tiny_vit.rs +++ b/candle-transformers/src/models/segment_anything/tiny_vit.rs @@ -28,7 +28,7 @@ impl Conv2dBN { impl Module for Conv2dBN { fn forward(&self, xs: &Tensor) -> Result<Tensor> { let _enter = self.span.enter(); - xs.apply(&self.c)?.apply(&self.bn) + xs.apply(&self.c)?.apply_t(&self.bn, false) } } |