Views
No views yet
tryon to trigger the image generation.1from diffusers import AutoPipelineForText2Image
2import torch
3
4pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-Kontext-dev', torch_dtype=torch.bfloat16).to('cuda')
5pipeline.load_lora_weights('amitkp621/AR-5-lora', weight_name='AR-5_000000250.safetensors')
6image = pipeline('tryon').images[0]
7image.save("my_image.png")