Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,11 +10,6 @@ device = "cuda"
|
|
| 10 |
model_repo_id = "anon4ik/noobaiXLNAIXL_epsilonPred05Version" # Replace to the model you would like to use
|
| 11 |
|
| 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)
|
| 19 |
|
| 20 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
|
| 10 |
model_repo_id = "anon4ik/noobaiXLNAIXL_epsilonPred05Version" # Replace to the model you would like to use
|
| 11 |
|
| 12 |
pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, torch_dtype=torch.float16)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
pipe = pipe.to(device)
|
| 14 |
|
| 15 |
MAX_SEED = np.iinfo(np.int32).max
|