Fallback if webui is an old version without sd_samplers_timesteps, #39

scripts/sd_save_intermediate_images.py

@@ -13,7 +13,13 @@
 from modules import script_callbacks
 from modules.processing import Processed, process_images, fix_seed, create_infotext
 from modules.sd_samplers_kdiffusion import KDiffusionSampler
-from modules.sd_samplers_timesteps import CompVisSampler as TimestepsSampler, samplers_timesteps
+try:
+    from modules.sd_samplers_timesteps import CompVisSampler as TimestepsSampler, samplers_timesteps
+except:
+    # Fallback if webui is an old version without sd_samplers_timesteps
+    class TimestepsSampler:
+        callback_state = None
+    from modules.sd_samplers_compvis import samplers_data_compvis as samplers_timesteps
 from modules.sd_samplers_common import sample_to_image
 from modules.images import save_image, FilenameGenerator, get_next_sequence_number
 from modules.shared import opts, state, cmd_opts