fix model name
This commit is contained in:
parent
b971b659e3
commit
260f7a29f4
|
@ -150,7 +150,7 @@ def upscale_stable_diffusion(ctx: ServerContext, params: UpscaleParams, image: I
|
||||||
generator = torch.manual_seed(0)
|
generator = torch.manual_seed(0)
|
||||||
seed = generator.initial_seed()
|
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(
|
upscale = lambda i: pipeline(
|
||||||
'an astronaut eating a hamburger',
|
'an astronaut eating a hamburger',
|
||||||
image=i,
|
image=i,
|
||||||
|
|
Loading…
Reference in New Issue