Add 1024 model (#5)

- Update img_size from 512 to 1024 (513bb83ac9505c3f19d0fa9e91f6d24d5385c7d8)
- Upload 1024x1024 model weights (993d7b6874f320d28067b0df28fc3697ec420546)

birefnet.py

@@ -53,7 +53,7 @@ class Config(PretrainedConfig):
             }[self.task]
         ][1]    # choose 0 to skip
         self.lr = (1e-4 if 'DIS5K' in self.task else 1e-5) * math.sqrt(self.batch_size / 4)     # DIS needs high lr to converge faster. Adapt the lr linearly
-        self.size = 512
+        self.size = 1024
         self.num_workers = max(4, self.batch_size)          # will be decrease to min(it, batch_size) at the initialization of the data_loader
 
         # Backbone settings

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:28416acdaedc3fbee0f80f888893a2fdca60ae81b1b27a41433a4e791598e330
+oid sha256:28f8acba2736067bf2eb8152f2d3ce75a388dd4e4e57f068b30760a8fe1c44d0
 size 884878824