Spaces:
Runtime error
Runtime error
import torch | |
from diffusers.utils import load_image | |
from diffusers import FluxControlNetPipeline, FluxControlNetModel | |
base_model = 'black-forest-labs/FLUX.1-dev' | |
controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Union' | |
controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16) | |
pipe = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16) | |
pipe.to("cuda") | |
control_image_canny = load_image("https://huggingface.co/InstantX/FLUX.1-dev-Controlnet-Union-alpha/resolve/main/images/canny.jpg") | |
controlnet_conditioning_scale = 0.5 | |
control_mode = 0 | |
width, height = control_image.size | |
prompt = 'A bohemian-style female travel blogger with sun-kissed skin and messy beach waves.' | |
image = pipe( | |
prompt, | |
control_image=control_image, | |
control_mode=control_mode, | |
width=width, | |
height=height, | |
controlnet_conditioning_scale=controlnet_conditioning_scale, | |
num_inference_steps=24, | |
guidance_scale=3.5, | |
).images[0] | |
image.save("image.jpg") | |