Update tiny_vit.py to fix bug
diff --git a/timm/models/tiny_vit.py b/timm/models/tiny_vit.py
index 4b5836584c..96a88db7f3 100644
--- a/timm/models/tiny_vit.py
+++ b/timm/models/tiny_vit.py
@@ -378,6 +378,7 @@ def __init__(
 
         super().__init__()
         self.depth = depth
+        self.out_dim =  out_dim
 
         # patch merging layer
         if downsample is not None:

There was a minor error when attempting to use TinyVit. I found an undeclared variable and have fixed it.