diff --git a/python/turbine_models/custom_models/sd_inference/utils.py b/python/turbine_models/custom_models/sd_inference/utils.py index 610331584..11fdaf19f 100644 --- a/python/turbine_models/custom_models/sd_inference/utils.py +++ b/python/turbine_models/custom_models/sd_inference/utils.py @@ -1,6 +1,7 @@ import numpy as np import safetensors + def save_external_weights( mapper, model, @@ -20,4 +21,4 @@ def save_external_weights( def largest_error(array1, array2): absolute_diff = np.abs(array1 - array2) max_error = np.max(absolute_diff) - return max_error \ No newline at end of file + return max_error