Spaces:

anzorq

/

sd-space-creator

like 67

Build error

Another feedback, your scheduler settings choice doesn't work with a v2 model.

scheduler = DPMSolverMultistepScheduler(
    beta_start=0.00085,
    beta_end=0.012,
    beta_schedule="scaled_linear",
    num_train_timesteps=1000,
    trained_betas=None,
    predict_epsilon=True,
    thresholding=False,
    algorithm_type="dpmsolver++",
    solver_type="midpoint",
    lower_order_final=True,
)

In my tests I'm loading the setting from the pre-trained model, maybe this is a more general solution ?

scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")