Dalladrain commited on
Commit
8088bfa
·
verified ·
1 Parent(s): fdf79cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,11 +10,10 @@ from PIL import Image, ImageDraw, ImageFont
10
  # Load Stable Diffusion with Euler Scheduler
11
  # =====================================================
12
  model_id = "stabilityai/stable-diffusion-2-1-base"
13
-
14
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
15
  pipe = StableDiffusionPipeline.from_pretrained(
16
  model_id, scheduler=scheduler, torch_dtype=torch.float16
17
- ).to("cuda") # or "cpu" if GPU not available
18
 
19
  # =====================================================
20
  # Comic Styles & Descriptions
 
10
  # Load Stable Diffusion with Euler Scheduler
11
  # =====================================================
12
  model_id = "stabilityai/stable-diffusion-2-1-base"
 
13
  scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  model_id, scheduler=scheduler, torch_dtype=torch.float16
16
+ ).to("cuda") # Change to "cpu" if GPU unavailable
17
 
18
  # =====================================================
19
  # Comic Styles & Descriptions