Spaces:
Thafx
/
Runtime error

Thafx commited on
Commit
9c78288
·
1 Parent(s): d92176e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline #,DPMSolverMultistepScheduler
2
  import gradio as gr
3
  import torch
4
  from PIL import Image
@@ -6,17 +6,17 @@ from PIL import Image
6
  model_id = 'Dunkindont/Foto-Assisted-Diffusion-FAD_V0'
7
  prefix = 'RAW photo,'
8
 
9
- # scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
14
- # scheduler=scheduler)
15
 
16
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
17
  model_id,
18
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
19
- # scheduler=scheduler)
20
 
21
  if torch.cuda.is_available():
22
  pipe = pipe.to("cuda")
 
1
+ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
2
  import gradio as gr
3
  import torch
4
  from PIL import Image
 
6
  model_id = 'Dunkindont/Foto-Assisted-Diffusion-FAD_V0'
7
  prefix = 'RAW photo,'
8
 
9
+ scheduler = DPMSolverMultistepScheduler.from_pretrained(model_id, folder="FAD768-Diffuser%20Model", subfolder="scheduler")
10
 
11
  pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
14
+ scheduler=scheduler)
15
 
16
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
17
  model_id,
18
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
19
+ scheduler=scheduler)
20
 
21
  if torch.cuda.is_available():
22
  pipe = pipe.to("cuda")