Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ def install_model(namemodel,tokenn,namemodelonxx):
|
|
9 |
def convert_to_onnx(model,namemodelonxx):
|
10 |
vocab_size = model.text_encoder.embed_tokens.weight.size(0)
|
11 |
example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
|
12 |
-
x=f"{
|
13 |
torch.onnx.export(
|
14 |
model, # The model to be exported
|
15 |
example_input, # Example input for the model
|
|
|
9 |
def convert_to_onnx(model,namemodelonxx):
|
10 |
vocab_size = model.text_encoder.embed_tokens.weight.size(0)
|
11 |
example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
|
12 |
+
x=f"{wasmdashai/namemodelonxx}.onnx"
|
13 |
torch.onnx.export(
|
14 |
model, # The model to be exported
|
15 |
example_input, # Example input for the model
|