https://github.com/bes-dev/story-diffusion-runpod-serverless-worker
StoryDiffusion serverless worker
https://github.com/bes-dev/story-diffusion-runpod-serverless-worker
Last synced: about 1 year ago
JSON representation
StoryDiffusion serverless worker
- Host: GitHub
- URL: https://github.com/bes-dev/story-diffusion-runpod-serverless-worker
- Owner: bes-dev
- License: apache-2.0
- Created: 2024-05-05T22:41:09.000Z (about 2 years ago)
- Default Branch: master
- Last Pushed: 2024-05-13T15:00:11.000Z (about 2 years ago)
- Last Synced: 2025-03-27T23:02:17.323Z (over 1 year ago)
- Language: Python
- Size: 43.9 KB
- Stars: 16
- Watchers: 1
- Forks: 7
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# StoryDiffusion: Serverless RunPod Worker
## RunPod Endpoint
This repository contains the worker for the StoryDiffusion AI Endpoints.
## Docker Image
```bash
docker build .
```
or
```bash
docker pull devbes/story-diffusion-serverless-worker:latest
```
## Environment Variables
### S3 storage
- BUCKET_ENDPOINT_URL
- BUCKET_ACCESS_KEY_ID
- BUCKET_SECRET_ACCESS_KEY
### Dockerfile configuration
- WORKER_MODEL_NAME (default = SG161222/RealVisXL_V4.0)
- WORKER_ID_LENGTH (default = 4)
- WORKER_TOTAL_LENGTH (default = 5)
- WORKER_SCHEDULER_TYPE (default = euler)
## Continuous Deployment
This worker follows a modified version of the [worker template](https://github.com/runpod-workers/worker-template) where the Docker build workflow contains additional SD models to be built and pushed.
## API
Use 'img' as a trigger word for personalized generation cases.
```json
{
"input": {
"prompts": [, ],
"negative_prompt": ,
"width": ,
"height": ,
"sa32": ,
"sa64": ,
"guidance_scale": ,
"num_inference_steps": ,
"seed": ,
"image_ref":
}
}
```
Sample request:
```json
{
"input": {
"prompts": ["Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. discovering a golden key in his grandmother's attic.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. talking to a squirrel in a magical forest.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. jumping on giant marshmallows at the top of a mountain.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. hopping into a boat on a sparkling river.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. talking to a mole in an underground cave.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. dancing at a village festival.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. talking to the squirrel again in the magical forest.", "Harold img is a curious and clever boy with bright blue eyes and messy brown hair. He always wears a red hat and carries a tiny backpack full of gadgets. telling his grandmother about his adventure at her home."],
"negative_prompt": "naked, deformed, bad anatomy, disfigured, poorly drawn face, mutation, extra limb, ugly, disgusting, poorly drawn hands, missing limb, floating limbs, disconnected limbs, blurry, watermarks, oversaturated, distorted hands, amputation",
"width": 768,
"height": 768,
"sa32": 0.5,
"sa64": 0.5,
"guidance_scale": 5.0,
"num_inference_steps": 25,
"seed": 42,
"image_ref": "https://alpinabook.ru/upload/resize_cache/iblock/8d9/550_800_1/8d9cd63476f15e85f0d8796555ab1e6b.jpg"
}
}
```
## Related Resources
This project is based on original implementation of [StoryDiffusion](https://github.com/HVision-NKU/StoryDiffusion).