fix: missing defaults

runpod-workers · Aug 8, 2023 · 859082f · 859082f
1 parent 6a1288e
commit 859082f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,10 +1,10 @@
 <div align="center">
 
-<h1>Template | Worker</h1>
+<h1>Stable Diffusion XL | Worker</h1>
 
 [![CI | Test Worker](https://github.com/runpod-workers/worker-template/actions/workflows/CI-test_worker.yml/badge.svg)](https://github.com/runpod-workers/worker-template/actions/workflows/CI-test_worker.yml)
 &nbsp;
-[![Docker Image](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml/badge.svg)](https://github.com/runpod-workers/worker-template/actions/workflows/CD-docker_dev.yml)
+[![CD | Dev Docker Image](https://github.com/runpod-workers/worker-sdxl/actions/workflows/CD-docker_dev.yml/badge.svg)](https://github.com/runpod-workers/worker-sdxl/actions/workflows/CD-docker_dev.yml)
 
 🚀 | A simple worker that can be used as a starting point to build your own custom RunPod Endpoint API worker.
 </div>

diff --git a/src/rp_schemas.py b/src/rp_schemas.py
@@ -6,14 +6,17 @@
     'negative_prompt': {
         'type': str,
         'required': False,
+        'default': None
     },
     'height': {
         'type': int,
         'required': False,
+        'default': 512
     },
     'width': {
         'type': int,
         'required': False,
+        'default': 512
     },
     'num_inference_steps': {
         'type': int,