Skip to content

Commit 5a46deb

Browse files
update dtype of timestamp
1 parent 127ec9a commit 5a46deb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4527,7 +4527,7 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int
45274527
shape = [self.batch_size, self.sequence_length, self.cap_feat_dim]
45284528
return self.random_float_tensor(shape, framework=framework, dtype=float_dtype)
45294529
if input_name == "timestep":
4530-
return self.random_int_tensor([1], max_value=20, min_value=1, framework=framework, dtype=int_dtype)
4530+
return self.random_float_tensor([1], max_value=1, min_value=0, framework=framework, dtype=float_dtype)
45314531

45324532
return super().generate(input_name, framework, int_dtype, float_dtype)
45334533

0 commit comments

Comments
 (0)