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 | ... | @@ -32,6 +32,8 @@ torch==2.0.0 |
| # ML dependencies | # ML dependencies | ||
| transformers>=4.35.0 | transformers>=4.35.0 | ||
| accelerate>=0.24.0 | accelerate>=0.24.0 | ||
| diffusers>=0.25.0 # For Stable Diffusion image generation | |||
| safetensors>=0.4.0 # Required by diffusers | |||
| # System resource detection | # System resource detection | ||
| psutil>=5.9.0 | psutil>=5.9.0 | ||
| ... | ... |
Please
register
or
sign in
to comment