Spaces:

omer11a
/

bounded-attention

Runtime error

omer11a commited on Mar 27, 2024

Commit

50494f3

1 Parent(s): e491328

Removed optimizations

Files changed (1) hide show

app.py CHANGED Viewed

@@ -194,8 +194,6 @@ def main():
     model_path = "stabilityai/stable-diffusion-xl-base-1.0"
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
     model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, torch_dtype=torch.float16).to(device)
-    model.unet.set_default_attn_processor()
-    model.enable_sequential_cpu_offload()
     nltk.download('averaged_perceptron_tagger')

     model_path = "stabilityai/stable-diffusion-xl-base-1.0"
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
     model = StableDiffusionXLPipeline.from_pretrained(model_path, scheduler=scheduler, torch_dtype=torch.float16).to(device)
     nltk.download('averaged_perceptron_tagger')