Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 from PIL import Image
 
-model_id = '22h/vintedois-diffusion-v0-1'
+model_id = '22h/vintedois-diffusion-v0-2'
 prefix = ''
      
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
@@ -12,11 +12,13 @@ pipe = StableDiffusionPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler)
+pipe.enable_attention_slicing(1)
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler)
+pipe_i2i.enable_attention_slicing(1)
 
 if torch.cuda.is_available():
   pipe = pipe.to("cuda")