Instructions to use skyadmin/cog-webui-sd with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use skyadmin/cog-webui-sd with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("skyadmin/cog-webui-sd", device_map="auto") - Notebooks
- Google Colab
- Kaggle
wolverinn commited on
Commit ·
7a1cf92
1
Parent(s): 92ba35f
add log
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -154,6 +154,7 @@ class EndpointHandler():
|
|
| 154 |
"seed": -1,
|
| 155 |
}
|
| 156 |
if "prompt" in data.keys():
|
|
|
|
| 157 |
args["prompt"] = data["prompt"]
|
| 158 |
p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **args)
|
| 159 |
processed = process_images(p)
|
|
|
|
| 154 |
"seed": -1,
|
| 155 |
}
|
| 156 |
if "prompt" in data.keys():
|
| 157 |
+
print("get prompt from request: ", data["prompt"])
|
| 158 |
args["prompt"] = data["prompt"]
|
| 159 |
p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **args)
|
| 160 |
processed = process_images(p)
|