Spaces:
Running
on
Zero
Running
on
Zero
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
|