Diffusers
English
stable-diffusion
stable-diffusion-diffusers
inpainting
art
artistic
anime
absolute-realism
Instructions to use diffusers/tools with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use diffusers/tools with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("diffusers/tools", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Commit ·
a1fd59a
1
Parent(s): fcc7a59
up
Browse files- run_xl_ediffi.py +0 -14
run_xl_ediffi.py
CHANGED
|
@@ -37,20 +37,6 @@ pipe_high_noise = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.
|
|
| 37 |
pipe_high_noise.scheduler = scheduler
|
| 38 |
pipe_high_noise.to("cuda")
|
| 39 |
|
| 40 |
-
scheduler = DPMSolverMultistepScheduler(
|
| 41 |
-
beta_start=0.00085,
|
| 42 |
-
beta_end=0.012,
|
| 43 |
-
beta_schedule="scaled_linear",
|
| 44 |
-
prediction_type="epsilon",
|
| 45 |
-
num_train_timesteps=1000,
|
| 46 |
-
trained_betas=None,
|
| 47 |
-
thresholding=False,
|
| 48 |
-
algorithm_type="dpmsolver++",
|
| 49 |
-
solver_type="midpoint",
|
| 50 |
-
lower_order_final=True,
|
| 51 |
-
use_karras_sigmas=True,
|
| 52 |
-
)
|
| 53 |
-
|
| 54 |
pipe_low_noise = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
|
| 55 |
pipe_low_noise.scheduler = scheduler
|
| 56 |
pipe_low_noise.to("cuda")
|
|
|
|
| 37 |
pipe_high_noise.scheduler = scheduler
|
| 38 |
pipe_high_noise.to("cuda")
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
pipe_low_noise = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
|
| 41 |
pipe_low_noise.scheduler = scheduler
|
| 42 |
pipe_low_noise.to("cuda")
|