mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
new sampler features
This commit is contained in:
@@ -8,6 +8,7 @@ from modules.tcd import TCDScheduler
|
||||
debug = shared.log.trace if os.environ.get('SD_SAMPLER_DEBUG', None) is not None else lambda *args, **kwargs: None
|
||||
debug('Trace: SAMPLER')
|
||||
|
||||
|
||||
try:
|
||||
from diffusers import (
|
||||
CMStochasticIterativeScheduler,
|
||||
@@ -35,6 +36,7 @@ except Exception as e:
|
||||
import diffusers
|
||||
shared.log.error(f'Diffusers import error: version={diffusers.__version__} error: {e}')
|
||||
|
||||
|
||||
config = {
|
||||
# beta_start, beta_end are typically per-scheduler, but we don't want them as they should be taken from the model itself as those are values model was trained on
|
||||
# prediction_type is ideally set in model as well, but it maybe needed that we do auto-detect of model type in the future
|
||||
|
||||
Reference in New Issue
Block a user