from diffusers import AutoPipelineForText2Image
import torch
pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
pipeline.load_lora_weights('None', weight_name='relax-piano.safetensors')
image = pipeline('This is a vibrant digital illustration featuring a serene forest scene. In the foreground, a tranquil pond reflects the surrounding lush greenery and a clear, blue sky with a few fluffy clouds. The pond is dotted with various-sized rocks and surrounded by vibrant flora, including orange and yellow flowers with large, glossy leaves. The water is calm, with gentle ripples and a few floating leaves. The central focus of the image is a cheerful, round, yellow sunfish-like creature with a smiling face, small pink flowers on its head, and closed eyes. It appears to be swimming contentedly in the pond, adding a whimsical and playful element to the scene. The background is a dense forest filled with tall, leafy trees with a mix of green and orange hues, suggesting autumn. The sunlight filters through the leaves, creating a dappled effect on the ground and the water. Small birds are seen flying or perched on the branches, adding to the peaceful ambiance. The overall style of the illustration is whimsical and cartoonish, with bright, bold colors and clean lines, emphasizing the tranquility and beauty of nature. The artist's attention to detail and the soft, almost dreamy lighting enhance the scene's charm.').images[0]
image.save("my_image.png")