Skip to content

Commit 40b20c6

Browse files
committed
Update pipeline_bria.py
1 parent 0f42410 commit 40b20c6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/pipelines/bria/pipeline_bria.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def encode_prompt(
239239
batch_size = prompt_embeds.shape[0]
240240

241241
if prompt_embeds is None:
242-
prompt_embeds = get_t5_prompt_embeds(
242+
prompt_embeds = self.get_t5_prompt_embeds(
243243
self.tokenizer,
244244
self.text_encoder,
245245
prompt=prompt,
@@ -249,7 +249,7 @@ def encode_prompt(
249249
).to(dtype=self.transformer.dtype)
250250

251251
if do_classifier_free_guidance and negative_prompt_embeds is None:
252-
if not is_ng_none(negative_prompt):
252+
if not self.is_ng_none(negative_prompt):
253253
negative_prompt = (
254254
batch_size * [negative_prompt] if isinstance(negative_prompt, str) else negative_prompt
255255
)
@@ -501,7 +501,7 @@ def __call__(
501501
self.scheduler, num_inference_steps, device, None, None
502502
)
503503
else:
504-
sigmas = get_original_sigmas(
504+
sigmas = self.get_original_sigmas(
505505
num_train_timesteps=self.scheduler.config.num_train_timesteps,
506506
num_inference_steps=num_inference_steps,
507507
)

0 commit comments

Comments
 (0)