Views
No views yet
1from diffusers import AutoPipelineForText2Image
2import torch
3
4device = "cuda" if torch.cuda.is_available() else "cpu"
5
6pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to(device)
7pipeline.load_lora_weights('Keltezaa/neon-environments', weight_name='Neon_Environments.safetensors')
8image = pipeline('Your custom prompt').images[0]