lanzhiwang commited on
Commit
5698580
1 Parent(s): 3995c81
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -31,12 +31,11 @@ def infer(prompt, negative, scale):
31
  model_id = "stabilityai/stable-diffusion-2"
32
  # Use the Euler scheduler here instead
33
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
34
- pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
35
  # pipe = pipe.to("cuda")
36
  images = pipe(prompt).images
37
  return images
38
 
39
-
40
  css = """
41
  .gradio-container {
42
  font-family: 'IBM Plex Sans', sans-serif;
 
31
  model_id = "stabilityai/stable-diffusion-2"
32
  # Use the Euler scheduler here instead
33
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
34
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float32)
35
  # pipe = pipe.to("cuda")
36
  images = pipe(prompt).images
37
  return images
38
 
 
39
  css = """
40
  .gradio-container {
41
  font-family: 'IBM Plex Sans', sans-serif;