Menyu commited on
Commit
5f83ce1
1 Parent(s): 418dbb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model_repo_id = "anon4ik/noobaiXLNAIXL_epsilonPred05Version" # Replace to the m
12
  pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float16)
13
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
14
  pipe.scheduler.register_to_config(
15
- prediction_type="diffusers",
16
  rescale_betas_zero_snr=True,
17
  )
18
  pipe = pipe.to(device)
 
12
  pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float16)
13
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
14
  pipe.scheduler.register_to_config(
15
+ prediction_type="epsilon",
16
  rescale_betas_zero_snr=True,
17
  )
18
  pipe = pipe.to(device)