Add diffusers and safetensors to requirements for image generation
diffusers is required for Stable Diffusion image generation
Showing
| ... | ... | @@ -32,6 +32,8 @@ torch==2.0.0 |
| # ML dependencies | ||
| transformers>=4.35.0 | ||
| accelerate>=0.24.0 | ||
| diffusers>=0.25.0 # For Stable Diffusion image generation | ||
| safetensors>=0.4.0 # Required by diffusers | ||
| # System resource detection | ||
| psutil>=5.9.0 | ||
| ... | ... |
Please
register
or
sign in
to comment