Skip to content

Commit 2a67cef

Browse files
committed
Address bot comment
1 parent 210031b commit 2a67cef

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

examples/scripts/interpolator.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ def interpolate_models(
2626
if output_model_path is None:
2727
output_model_path = f"{trained_model_path}_interp"
2828

29+
if not (0.0 <= trained_model_weight <= 1.0):
30+
raise ValueError(f"trained_model_weight must be in [0,1], got {trained_model_weight}")
31+
2932
model_kwargs = {}
3033
if torch_dtype is not None:
3134
if isinstance(torch_dtype, str):

0 commit comments

Comments
 (0)