Unverified Commit 410b61ad authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Revert (for now) the change on `Deta` in #22437 (#22750)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 9dfd6a4b
...@@ -244,7 +244,7 @@ class DetaObjectDetectionOutput(ModelOutput): ...@@ -244,7 +244,7 @@ class DetaObjectDetectionOutput(ModelOutput):
def _get_clones(module, N): def _get_clones(module, N):
return nn.ModuleList([module for i in range(N)]) return nn.ModuleList([copy.deepcopy(module) for i in range(N)])
def inverse_sigmoid(x, eps=1e-5): def inverse_sigmoid(x, eps=1e-5):
......
...@@ -192,6 +192,10 @@ class DetaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin ...@@ -192,6 +192,10 @@ class DetaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin
return False return False
@unittest.skip("Skip for now. PR #22437 causes some loading issue. See (not merged) #22656 for some discussions.")
def test_can_use_safetensors(self):
super().test_can_use_safetensors()
# special case for head models # special case for head models
def _prepare_for_class(self, inputs_dict, model_class, return_labels=False): def _prepare_for_class(self, inputs_dict, model_class, return_labels=False):
inputs_dict = super()._prepare_for_class(inputs_dict, model_class, return_labels=return_labels) inputs_dict = super()._prepare_for_class(inputs_dict, model_class, return_labels=return_labels)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment