apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix model name

This commit is contained in:
Sean Sube 2023-01-26 23:15:43 -06:00
parent b971b659e3
commit 260f7a29f4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/upscale.py
View File

@ -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,