Update convert.py
Browse files- convert.py +0 -1
convert.py
CHANGED
|
@@ -183,7 +183,6 @@ def check_final_model(model_id: str, folder: str, token: Optional[str]):
|
|
| 183 |
pixel_values = torch.randn(1, 3, 224, 224)
|
| 184 |
input_values = torch.arange(1000).float().unsqueeze(0)
|
| 185 |
kwargs = {}
|
| 186 |
-
import ipdb;ipdb.set_trace()
|
| 187 |
if "input_ids" in sig.parameters:
|
| 188 |
kwargs["input_ids"] = input_ids
|
| 189 |
if "decoder_input_ids" in sig.parameters:
|
|
|
|
| 183 |
pixel_values = torch.randn(1, 3, 224, 224)
|
| 184 |
input_values = torch.arange(1000).float().unsqueeze(0)
|
| 185 |
kwargs = {}
|
|
|
|
| 186 |
if "input_ids" in sig.parameters:
|
| 187 |
kwargs["input_ids"] = input_ids
|
| 188 |
if "decoder_input_ids" in sig.parameters:
|