From 260f7a29f44e9426ee9a1665443a2a6240b67304 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 26 Jan 2023 23:15:43 -0600 Subject: [PATCH] fix model name --- api/onnx_web/upscale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py index a1df5c7f..4c26d027 100644 --- a/api/onnx_web/upscale.py +++ b/api/onnx_web/upscale.py @@ -150,7 +150,7 @@ def upscale_stable_diffusion(ctx: ServerContext, params: UpscaleParams, image: I generator = torch.manual_seed(0) seed = generator.initial_seed() - pipeline = StableDiffusionUpscalePipeline.from_pretrained('stabilityai/stable-diffusion-x4-upscaling') + pipeline = StableDiffusionUpscalePipeline.from_pretrained('stabilityai/stable-diffusion-x4-upscaler') upscale = lambda i: pipeline( 'an astronaut eating a hamburger', image=i,