scheduler fixes

2025-01-10 00:36:58 +02:00
parent 11ee15a91e
commit 836b26aff3
1 changed files with 4 additions and 0 deletions
@@ -1177,6 +1177,10 @@ class HyVideoSampler:
        scheduler_config["flow_shift"] = flow_shift
        if scheduler == "SDE-DPMSolverMultistepScheduler":
            scheduler_config["algorithm_type"] = "sde-dpmsolver++"
+        elif scheduler == "SASolverScheduler":
+            scheduler_config["algorithm_type"] = "data_prediction"
+        else:
+            scheduler_config.pop("algorithm_type", None)
        #model["scheduler_config"]["use_beta_flow_sigmas"] = True
        
        noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config)