Matt commited on
Commit
1e8c51f
·
1 Parent(s): bef74d7

Tie weights correctly and update weights

Browse files
Files changed (2) hide show
  1. model.safetensors +2 -2
  2. modeling_florence2.py +6 -0
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a0931732cefbda398920d6a13bfc7eefc6c457a1463a17c7f4930b68b74b46d5
3
- size 542001482
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43c6d08c6afba4b18c373b5c0f94be59550c84ffe17993c1d3c2fe06c46f94ea
3
+ size 463221938
modeling_florence2.py CHANGED
@@ -2066,6 +2066,12 @@ class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel
2066
  # Initialize weights and apply final processing
2067
  self.post_init()
2068
 
 
 
 
 
 
 
2069
  def get_encoder(self):
2070
  return self.model.get_encoder()
2071
 
 
2066
  # Initialize weights and apply final processing
2067
  self.post_init()
2068
 
2069
+ def _tie_weights(self):
2070
+ if self.config.tie_word_embeddings:
2071
+ self._tie_or_clone_weights(self.model.encoder.embed_tokens, self.model.shared)
2072
+ self._tie_or_clone_weights(self.model.decoder.embed_tokens, self.model.shared)
2073
+ self._tie_or_clone_weights(self.lm_head, self.model.shared)
2074
+
2075
  def get_encoder(self):
2076
  return self.model.get_encoder()
2077