Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -97,7 +97,7 @@ class OnnxModelConverter:
|
|
97 |
onnx_file = f"/tmp/{onnx_filename}.onnx"
|
98 |
|
99 |
vocab_size = model.text_encoder.embed_tokens.weight.size(0)
|
100 |
-
example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.
|
101 |
|
102 |
torch.onnx.export(
|
103 |
model,
|
|
|
97 |
onnx_file = f"/tmp/{onnx_filename}.onnx"
|
98 |
|
99 |
vocab_size = model.text_encoder.embed_tokens.weight.size(0)
|
100 |
+
example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
|
101 |
|
102 |
torch.onnx.export(
|
103 |
model,
|