Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,6 @@ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1
|
|
| 11 |
base = DiffusionPipeline.from_pretrained(
|
| 12 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 13 |
)
|
| 14 |
-
base.to("cuda")
|
| 15 |
refiner = DiffusionPipeline.from_pretrained(
|
| 16 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
| 17 |
text_encoder_2=base.text_encoder_2,
|
|
@@ -20,7 +19,6 @@ refiner = DiffusionPipeline.from_pretrained(
|
|
| 20 |
use_safetensors=True,
|
| 21 |
variant="fp16",
|
| 22 |
)
|
| 23 |
-
refiner.to("cuda")
|
| 24 |
|
| 25 |
# Define how many steps and what % of steps to be run on each experts (80/20) here
|
| 26 |
n_steps = 40
|
|
|
|
| 11 |
base = DiffusionPipeline.from_pretrained(
|
| 12 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 13 |
)
|
|
|
|
| 14 |
refiner = DiffusionPipeline.from_pretrained(
|
| 15 |
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
| 16 |
text_encoder_2=base.text_encoder_2,
|
|
|
|
| 19 |
use_safetensors=True,
|
| 20 |
variant="fp16",
|
| 21 |
)
|
|
|
|
| 22 |
|
| 23 |
# Define how many steps and what % of steps to be run on each experts (80/20) here
|
| 24 |
n_steps = 40
|