nan2 commited on
Commit
8757d09
·
1 Parent(s): 30dc91b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import torch
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
6
- #model_id = "models/nan2/lcbanner"
7
- pipe = StableDiffusionPipeline.from_pretrained("models/nan2/lcbanner", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2
 
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
5
 
6
+ #model_id = "nan2/lcbanner"
7
+ pipe = StableDiffusionPipeline.from_pretrained("nan2/lcbanner", torch_type=torch.float16, revision="fp16")
8
  pipe = pipe.to("cuda")
9
  #torch.backends.cudnn.benchmark = True
10
  num_samples = 2