Spaces:
Sleeping
Sleeping
Fix tiny image model loading by removing safetensors-only flag
Browse files
tiny_image_gen/service.py
CHANGED
|
@@ -39,7 +39,6 @@ class TinyImageService:
|
|
| 39 |
safety_checker=None,
|
| 40 |
requires_safety_checker=False,
|
| 41 |
low_cpu_mem_usage=True,
|
| 42 |
-
use_safetensors=True,
|
| 43 |
)
|
| 44 |
pipe = pipe.to(DEVICE)
|
| 45 |
pipe.enable_attention_slicing()
|
|
|
|
| 39 |
safety_checker=None,
|
| 40 |
requires_safety_checker=False,
|
| 41 |
low_cpu_mem_usage=True,
|
|
|
|
| 42 |
)
|
| 43 |
pipe = pipe.to(DEVICE)
|
| 44 |
pipe.enable_attention_slicing()
|