We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 210031b commit 2a67cefCopy full SHA for 2a67cef
examples/scripts/interpolator.py
@@ -26,6 +26,9 @@ def interpolate_models(
26
if output_model_path is None:
27
output_model_path = f"{trained_model_path}_interp"
28
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
+
32
model_kwargs = {}
33
if torch_dtype is not None:
34
if isinstance(torch_dtype, str):
0 commit comments