Doesn't Work
#6
by
qpqpqpqpqpqp - opened
Failed to import diffusers.pipelines.prx.pipeline_prx because of the following error (look up to see its traceback):
Could not import module 'AutoTokenizer'. Are this object's requirements defined correctly?
I just tried with the last version of diffusers (0.36) on Google colab and it worked.
I did need to install ftfy but I don't think it's related with PRX.
Do you have the last version of diffusers and transformers installed?
Here is a minimal code that worked on colab:
!pip install diffusers --upgrade
!pip install ftfy
from diffusers.pipelines.prx import PRXPipeline
import torch
# Load pipeline - VAE and text encoder will be loaded from HuggingFace
pipe = PRXPipeline.from_pretrained("Photoroom/prx-512-t2i-sft", torch_dtype=torch.bfloat16)
pipe.to("cuda")
prompt = "A front-facing portrait of a lion the golden savanna at sunset."
image = pipe(prompt, num_inference_steps=28, guidance_scale=5.0).images[0]
image.save("prx_output.png")
+uv pip install huggingface-hub==0.24.0 NumPy==1.26.4
Solved
qpqpqpqpqpqp changed discussion status to
closed