Remove type annotation in Siglip Attention Module (#38503)

* Remove type annotation

* remove print statement
This commit is contained in:
Yaswanth Gali
2025-06-02 21:21:07 +05:30
committed by GitHub
parent afb35a10ed
commit 1094dd34f7
3 changed files with 2 additions and 3 deletions

View File

@@ -739,7 +739,6 @@ class ModelTesterMixin:
model = model_class(config)
model.to(torch_device)
model.eval()
print(model_class)
with torch.no_grad():
first = model(**self._prepare_for_class(inputs_dict, model_class))[0]
second = model(**self._prepare_for_class(inputs_dict, model_class))[0]