Ahmed Faiyaz commited on
Commit
df9a2a8
1 Parent(s): 0b3569f

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -104,7 +104,7 @@ class MukhOboyobPipeline(DiffusionPipeline, TextualInversionLoaderMixin, LoraLoa
104
  ):
105
  super().__init__()
106
  text_encoder=AutoModel.from_pretrained("csebuetnlp/banglabert")
107
- tokenizer=AutoModel.from_pretrained("csebuetnlp/banglabert")
108
 
109
  if hasattr(scheduler.config, "steps_offset") and scheduler.config.steps_offset != 1:
110
  deprecation_message = (
 
104
  ):
105
  super().__init__()
106
  text_encoder=AutoModel.from_pretrained("csebuetnlp/banglabert")
107
+ tokenizer=AutoTokenizer.from_pretrained("csebuetnlp/banglabert")
108
 
109
  if hasattr(scheduler.config, "steps_offset") and scheduler.config.steps_offset != 1:
110
  deprecation_message = (