DeepFloyd

/

IF-I-XL-v1.0

like 593

When I use demo code:
#stage 2
image = stage_2(
image=image,
prompt_embeds=prompt_embeds,
negative_prompt_embeds=negative_embeds,
generator=generator,
output_type="pt",
).images

It throwed an exception like:
image = stage_2(
File "/usr/local/anaconda/envs/vision/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
TypeError: call() got an unexpected keyword argument 'image'

How could I do with it? I use diffusers 0.16.0 and pytrch 1.10.1