Update app.py

app.py

@@ -46,7 +46,10 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
             torch_dtype = torch.float32
 
         pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
-        pipe = pipe.to(device)
+        #pipe = pipe.to(device)
+        pipe.enable_model_cpu_offload()
+        pipe.vae.enable_slicing()
+        pipe.vae.enable_tiling()
         torch.cuda.empty_cache()
         image = pipe(prompt=Prompt, negative_prompt=negative_prompt, guidance_scale=0.0, num_inference_steps=4, width=1024, height=1024, generator=generator,).images[0]
         torch.cuda.empty_cache()