Spaces:

nan2
/

nan2-lcbanner

Runtime error

nan2 commited on Mar 24, 2023

Commit

84f02e7

1 Parent(s): db4a9fb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline
 #model_id = "hakurei/waifu-diffusion"
-pipe = StableDiffusionPipeline.from_pretrained("nan2/lcbanner", torch_type=torch.float16, revision="lcbanner")
 pipe = pipe.to("cuda")
 #torch.backends.cudnn.benchmark = True
 num_samples = 2

 from diffusers import StableDiffusionPipeline
 #model_id = "hakurei/waifu-diffusion"
+pipe = StableDiffusionPipeline.from_pretrained("nan2/lcbanner", torch_type=torch.float16)
 pipe = pipe.to("cuda")
 #torch.backends.cudnn.benchmark = True
 num_samples = 2