deleom commited on
Commit
f2c930a
·
1 Parent(s): 77b8131

Update edit_app.py

Browse files
Files changed (1) hide show
  1. 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(