hysts HF staff commited on
Commit
89efab8
1 Parent(s): 050ca2a
Files changed (1) hide show
  1. model.py +24 -0
model.py CHANGED
@@ -140,6 +140,8 @@ class Model:
140
  low_threshold: int,
141
  high_threshold: int,
142
  ) -> list[PIL.Image.Image]:
 
 
143
  if image_resolution > MAX_IMAGE_RESOLUTION:
144
  raise ValueError
145
  if num_images > MAX_NUM_IMAGES:
@@ -179,6 +181,8 @@ class Model:
179
  value_threshold: float,
180
  distance_threshold: float,
181
  ) -> list[PIL.Image.Image]:
 
 
182
  if image_resolution > MAX_IMAGE_RESOLUTION:
183
  raise ValueError
184
  if num_images > MAX_NUM_IMAGES:
@@ -219,6 +223,8 @@ class Model:
219
  seed: int,
220
  preprocessor_name: str,
221
  ) -> list[PIL.Image.Image]:
 
 
222
  if image_resolution > MAX_IMAGE_RESOLUTION:
223
  raise ValueError
224
  if num_images > MAX_NUM_IMAGES:
@@ -269,6 +275,8 @@ class Model:
269
  guidance_scale: float,
270
  seed: int,
271
  ) -> list[PIL.Image.Image]:
 
 
272
  if image_resolution > MAX_IMAGE_RESOLUTION:
273
  raise ValueError
274
  if num_images > MAX_NUM_IMAGES:
@@ -306,6 +314,8 @@ class Model:
306
  seed: int,
307
  preprocessor_name: str,
308
  ) -> list[PIL.Image.Image]:
 
 
309
  if image_resolution > MAX_IMAGE_RESOLUTION:
310
  raise ValueError
311
  if num_images > MAX_NUM_IMAGES:
@@ -362,6 +372,8 @@ class Model:
362
  seed: int,
363
  preprocessor_name: str,
364
  ) -> list[PIL.Image.Image]:
 
 
365
  if image_resolution > MAX_IMAGE_RESOLUTION:
366
  raise ValueError
367
  if num_images > MAX_NUM_IMAGES:
@@ -406,6 +418,8 @@ class Model:
406
  seed: int,
407
  preprocessor_name: str,
408
  ) -> list[PIL.Image.Image]:
 
 
409
  if image_resolution > MAX_IMAGE_RESOLUTION:
410
  raise ValueError
411
  if num_images > MAX_NUM_IMAGES:
@@ -449,6 +463,8 @@ class Model:
449
  seed: int,
450
  preprocessor_name: str,
451
  ) -> list[PIL.Image.Image]:
 
 
452
  if image_resolution > MAX_IMAGE_RESOLUTION:
453
  raise ValueError
454
  if num_images > MAX_NUM_IMAGES:
@@ -492,6 +508,8 @@ class Model:
492
  seed: int,
493
  preprocessor_name: str,
494
  ) -> list[PIL.Image.Image]:
 
 
495
  if image_resolution > MAX_IMAGE_RESOLUTION:
496
  raise ValueError
497
  if num_images > MAX_NUM_IMAGES:
@@ -535,6 +553,8 @@ class Model:
535
  seed: int,
536
  preprocessor_name: str,
537
  ) -> list[PIL.Image.Image]:
 
 
538
  if image_resolution > MAX_IMAGE_RESOLUTION:
539
  raise ValueError
540
  if num_images > MAX_NUM_IMAGES:
@@ -589,6 +609,8 @@ class Model:
589
  seed: int,
590
  preprocessor_name: str,
591
  ) -> list[PIL.Image.Image]:
 
 
592
  if image_resolution > MAX_IMAGE_RESOLUTION:
593
  raise ValueError
594
  if num_images > MAX_NUM_IMAGES:
@@ -629,6 +651,8 @@ class Model:
629
  guidance_scale: float,
630
  seed: int,
631
  ) -> list[PIL.Image.Image]:
 
 
632
  if image_resolution > MAX_IMAGE_RESOLUTION:
633
  raise ValueError
634
  if num_images > MAX_NUM_IMAGES:
 
140
  low_threshold: int,
141
  high_threshold: int,
142
  ) -> list[PIL.Image.Image]:
143
+ if image is None:
144
+ raise ValueError
145
  if image_resolution > MAX_IMAGE_RESOLUTION:
146
  raise ValueError
147
  if num_images > MAX_NUM_IMAGES:
 
181
  value_threshold: float,
182
  distance_threshold: float,
183
  ) -> list[PIL.Image.Image]:
184
+ if image is None:
185
+ raise ValueError
186
  if image_resolution > MAX_IMAGE_RESOLUTION:
187
  raise ValueError
188
  if num_images > MAX_NUM_IMAGES:
 
223
  seed: int,
224
  preprocessor_name: str,
225
  ) -> list[PIL.Image.Image]:
226
+ if image is None:
227
+ raise ValueError
228
  if image_resolution > MAX_IMAGE_RESOLUTION:
229
  raise ValueError
230
  if num_images > MAX_NUM_IMAGES:
 
275
  guidance_scale: float,
276
  seed: int,
277
  ) -> list[PIL.Image.Image]:
278
+ if image_and_mask is None:
279
+ raise ValueError
280
  if image_resolution > MAX_IMAGE_RESOLUTION:
281
  raise ValueError
282
  if num_images > MAX_NUM_IMAGES:
 
314
  seed: int,
315
  preprocessor_name: str,
316
  ) -> list[PIL.Image.Image]:
317
+ if image is None:
318
+ raise ValueError
319
  if image_resolution > MAX_IMAGE_RESOLUTION:
320
  raise ValueError
321
  if num_images > MAX_NUM_IMAGES:
 
372
  seed: int,
373
  preprocessor_name: str,
374
  ) -> list[PIL.Image.Image]:
375
+ if image is None:
376
+ raise ValueError
377
  if image_resolution > MAX_IMAGE_RESOLUTION:
378
  raise ValueError
379
  if num_images > MAX_NUM_IMAGES:
 
418
  seed: int,
419
  preprocessor_name: str,
420
  ) -> list[PIL.Image.Image]:
421
+ if image is None:
422
+ raise ValueError
423
  if image_resolution > MAX_IMAGE_RESOLUTION:
424
  raise ValueError
425
  if num_images > MAX_NUM_IMAGES:
 
463
  seed: int,
464
  preprocessor_name: str,
465
  ) -> list[PIL.Image.Image]:
466
+ if image is None:
467
+ raise ValueError
468
  if image_resolution > MAX_IMAGE_RESOLUTION:
469
  raise ValueError
470
  if num_images > MAX_NUM_IMAGES:
 
508
  seed: int,
509
  preprocessor_name: str,
510
  ) -> list[PIL.Image.Image]:
511
+ if image is None:
512
+ raise ValueError
513
  if image_resolution > MAX_IMAGE_RESOLUTION:
514
  raise ValueError
515
  if num_images > MAX_NUM_IMAGES:
 
553
  seed: int,
554
  preprocessor_name: str,
555
  ) -> list[PIL.Image.Image]:
556
+ if image is None:
557
+ raise ValueError
558
  if image_resolution > MAX_IMAGE_RESOLUTION:
559
  raise ValueError
560
  if num_images > MAX_NUM_IMAGES:
 
609
  seed: int,
610
  preprocessor_name: str,
611
  ) -> list[PIL.Image.Image]:
612
+ if image is None:
613
+ raise ValueError
614
  if image_resolution > MAX_IMAGE_RESOLUTION:
615
  raise ValueError
616
  if num_images > MAX_NUM_IMAGES:
 
651
  guidance_scale: float,
652
  seed: int,
653
  ) -> list[PIL.Image.Image]:
654
+ if image is None:
655
+ raise ValueError
656
  if image_resolution > MAX_IMAGE_RESOLUTION:
657
  raise ValueError
658
  if num_images > MAX_NUM_IMAGES: