Update app.py

app.py

@@ -9,7 +9,7 @@ import torch
 from torch import autocast
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
-model_id = 'Linaqruf/animagine-xl'
+model_id = 'aipicasso/picasso-diffusion-2-1'
 auth_token=os.environ["ACCESS_TOKEN"]
 
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler", use_auth_token=auth_token)
@@ -28,7 +28,6 @@ pipe_i2i = StableDiffusionXLImg2ImgPipeline.from_pretrained(
 )
 
 pipe=pipe.to("cuda")
-pipe.enable_model_cpu_offload()
 pipe_i2i=pipe_i2i.to("cuda")