Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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="
|
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)
|