AttributeError: 'VisCpmBeeTokenizer' object has no attribute 'encoder'
Zeqiang-Lai opened this issue · 0 comments
Zeqiang-Lai commented
Dear authors, I meet AttributeError: 'VisCpmBeeTokenizer' object has no attribute 'encoder' when running the following code:
#!/usr/bin/env python
# encoding: utf-8
from diffusers import DiffusionPipeline
from transformers import AutoModel
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('openbmb/VisCPM-Paint', trust_remote_code=True)
text_encoder = AutoModel.from_pretrained('openbmb/VisCPM-Paint', trust_remote_code=True)
print('load pipeline')
pipeline = DiffusionPipeline.from_pretrained('openbmb/VisCPM-Paint', custom_pipeline="openbmb/VisCPM-Paint", text_encoder=text_encoder, tokenizer=tokenizer)
pipeline = pipeline.to('cuda')
prompt = "a photo of an astronaut riding a horse on mars"
image = pipeline(prompt).images[0]
image.save("astronaut_rides_horse.png")