Spaces:
Runtime error
Runtime error
Update edit_app.py
Browse files- edit_app.py +1 -10
edit_app.py
CHANGED
|
@@ -53,15 +53,6 @@ device = "cuda"
|
|
| 53 |
base = DiffusionPipeline.from_pretrained(
|
| 54 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 55 |
).to(device)
|
| 56 |
-
refiner = DiffusionPipeline.from_pretrained(
|
| 57 |
-
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
| 58 |
-
text_encoder_2=base.text_encoder_2,
|
| 59 |
-
vae=base.vae,
|
| 60 |
-
torch_dtype=torch.float16,
|
| 61 |
-
use_safetensors=True,
|
| 62 |
-
variant="fp16",
|
| 63 |
-
)
|
| 64 |
-
refiner.to(device)
|
| 65 |
|
| 66 |
def main():
|
| 67 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_auth_token=auth_token, safety_checker=None).to("cuda")
|
|
@@ -132,7 +123,7 @@ def main():
|
|
| 132 |
image = base(
|
| 133 |
prompt=instruction,
|
| 134 |
num_inference_steps=steps,
|
| 135 |
-
denoising_end=high_noise_frac,
|
| 136 |
#output_type="latent",
|
| 137 |
).images[0]
|
| 138 |
#stable_created_image = refiner(
|
|
|
|
| 53 |
base = DiffusionPipeline.from_pretrained(
|
| 54 |
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
|
| 55 |
).to(device)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
def main():
|
| 58 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_auth_token=auth_token, safety_checker=None).to("cuda")
|
|
|
|
| 123 |
image = base(
|
| 124 |
prompt=instruction,
|
| 125 |
num_inference_steps=steps,
|
| 126 |
+
#denoising_end=high_noise_frac,
|
| 127 |
#output_type="latent",
|
| 128 |
).images[0]
|
| 129 |
#stable_created_image = refiner(
|