Skip to content

Commit 0f42410

Browse files
committed
Update pipeline_bria.py
1 parent 25cf206 commit 0f42410

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/diffusers/pipelines/bria/pipeline_bria.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def __init__(
118118
self.vae.to(dtype=torch.float32)
119119

120120
def get_t5_prompt_embeds(
121+
self,
121122
tokenizer: T5TokenizerFast,
122123
text_encoder: T5EncoderModel,
123124
prompt: Union[str, List[str]] = None,
@@ -169,15 +170,15 @@ def get_t5_prompt_embeds(
169170
return prompt_embeds
170171

171172
# in order the get the same sigmas as in training and sample from them
172-
def get_original_sigmas(num_train_timesteps=1000, num_inference_steps=1000):
173+
def get_original_sigmas(self,num_train_timesteps=1000, num_inference_steps=1000):
173174
timesteps = np.linspace(1, num_train_timesteps, num_train_timesteps, dtype=np.float32)[::-1].copy()
174175
sigmas = timesteps / num_train_timesteps
175176

176177
inds = [int(ind) for ind in np.linspace(0, num_train_timesteps - 1, num_inference_steps)]
177178
new_sigmas = sigmas[inds]
178179
return new_sigmas
179180

180-
def is_ng_none(negative_prompt):
181+
def is_ng_none(self,negative_prompt):
181182
return (
182183
negative_prompt is None
183184
or negative_prompt == ""

0 commit comments

Comments
 (0)