from diffusers import StableDiffusionPipeline
import torch

def execute_nvidia(prompt, filename):
    DEVICE = 'cuda'
    model_id = "runwayml/stable-diffusion-v1-5"
    pipe = StableDiffusionPipeline.from_pretrained(
        model_id,
        use_auth_token=True,
        torch_dtype=torch.float16
    ).to(DEVICE)
    pipe.enable_attention_slicing()
    image = pipe(prompt).images[0]
    image.save("pictures/"+ filename +".png")