diff --git a/html/reference.json b/html/reference.json index 479869762..b5113b612 100644 --- a/html/reference.json +++ b/html/reference.json @@ -1043,7 +1043,7 @@ "extras": "" }, "NoobAI XL 1.0 V-Pred": { - "path": "noobaiXLNAIXL_vPred10Version.safetensors@https://civitai.com/api/download/models/1190596", + "path": "noobaiXLNAIXL_vPred10Version.safetensors@https://huggingface.co/Laxhar/noobai-XL-Vpred-1.0/resolve/main/NoobAI-XL-Vpred-v1.0.safetensors", "preview": "noobaiXLNAIXL_vPred10Version.jpg", "desc": "", "tags": "community", @@ -1052,7 +1052,7 @@ "extras": "" }, "NoobAI XL 1.1 Epsilon": { - "path": "noobaiXLNAIXL_epsilonPred11Version.safetensors@https://civitai.com/api/download/models/1116447", + "path": "noobaiXLNAIXL_epsilonPred11Version.safetensors@https://huggingface.co/Laxhar/noobai-XL-1.1/resolve/main/NoobAI-XL-v1.1.safetensors", "preview": "noobaiXLNAIXL_epsilonPred11Version.jpg", "desc": "", "tags": "community", diff --git a/modules/sd_models.py b/modules/sd_models.py index 06e1d2b8e..f6313e371 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -591,7 +591,7 @@ def set_overrides(sd_model, checkpoint_info, model_type): scheduler_config['prediction_type'] = 'v_prediction' scheduler_config['beta_schedule'] = 'scaled_linear' scheduler_config['rescale_betas_zero_snr'] = True - sd_model.scheduler = diffusers.EulerDiscreteScheduler.from_config(scheduler_config) + sd_model.scheduler = diffusers.EulerAncestralDiscreteScheduler.from_config(scheduler_config) shared.log.info(f'Setting override: model="{checkpoint_info.name}" component=scheduler prediction="v-prediction" rescale=True') else: try: @@ -604,7 +604,7 @@ def set_overrides(sd_model, checkpoint_info, model_type): scheduler_config['beta_schedule'] = 'scaled_linear' if 'ztsnr' in keys: scheduler_config['rescale_betas_zero_snr'] = True - sd_model.scheduler = diffusers.EulerDiscreteScheduler.from_config(scheduler_config) + sd_model.scheduler = diffusers.EulerAncestralDiscreteScheduler.from_config(scheduler_config) shared.log.info(f'Setting override: model="{checkpoint_info.name}" component=scheduler prediction="v-prediction" rescale={scheduler_config.get("rescale_betas_zero_snr", False)}') except Exception as e: shared.log.debug(f'Setting override from keys failed: {e}')