process stage2 error
#19
by colacool - opened
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
I choose the stage2 model by mistake..
colacool changed discussion status to closed