Xenova HF staff commited on
Commit
992dd4d
·
verified ·
1 Parent(s): 38feeb2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -70,7 +70,7 @@ torch.onnx.export(
70
  },
71
  )
72
 
73
- dummy_decoder_inputs = torch.randint(100, (4, 32, 91))
74
  torch.onnx.export(
75
  decoder,
76
  dummy_decoder_inputs,
 
70
  },
71
  )
72
 
73
+ dummy_decoder_inputs = torch.randint(100, (4, model.config.num_quantizers, 91))
74
  torch.onnx.export(
75
  decoder,
76
  dummy_decoder_inputs,