@@ -227,7 +227,7 @@ def decode_latents(self, latents):
227
227
image = image .cpu ().permute (0 , 2 , 3 , 1 ).float ().numpy ()
228
228
return image
229
229
230
- def check_inputs (self , prompt , image , noise_level , callback_steps ):
230
+ def check_inputs (self , prompt , image , callback_steps ):
231
231
if not isinstance (prompt , str ) and not isinstance (prompt , list ):
232
232
raise ValueError (f"`prompt` has to be of type `str` or `list` but is { type (prompt )} " )
233
233
@@ -285,7 +285,6 @@ def __call__(
285
285
image : Union [torch .FloatTensor , PIL .Image .Image , List [PIL .Image .Image ]],
286
286
num_inference_steps : int = 75 ,
287
287
guidance_scale : float = 9.0 ,
288
- noise_level : int = 0 ,
289
288
negative_prompt : Optional [Union [str , List [str ]]] = None ,
290
289
generator : Optional [Union [torch .Generator , List [torch .Generator ]]] = None ,
291
290
latents : Optional [torch .FloatTensor ] = None ,
@@ -386,7 +385,7 @@ def __call__(
386
385
"""
387
386
388
387
# 1. Check inputs
389
- self .check_inputs (prompt , image , noise_level , callback_steps )
388
+ self .check_inputs (prompt , image , callback_steps )
390
389
391
390
# 2. Define call parameters
392
391
batch_size = 1 if isinstance (prompt , str ) else len (prompt )
0 commit comments