Spaces:

omer11a
/

bounded-attention

Runtime error

omer11a commited on Apr 3, 2024

Commit

431d204

1 Parent(s): c89d950

Fixed errors

Files changed (1) hide show

app.py CHANGED Viewed

@@ -256,7 +256,7 @@ def main():
     nltk.download("averaged_perceptron_tagger")
     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_attn_processor(AttnProcessor2_0())
     model.enable_sequential_cpu_offload()

     nltk.download("averaged_perceptron_tagger")
     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)
     model.unet.set_attn_processor(AttnProcessor2_0())
     model.enable_sequential_cpu_offload()