KailashV commited on
Commit
db077f4
·
1 Parent(s): f91d3b8

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,8 +1,7 @@
1
  import gradio as gr
2
  import spaces
3
  import torch
4
- import diffusers
5
- from diffusers import StableDiffusionXLPipeline,DiffusionPipeline,AutoPipelineForText2Image, DEISMultistepScheduler
6
 
7
  pipe = AutoPipelineForText2Image.from_pretrained('Lykon/AAM_XL_AnimeMix', torch_dtype=torch.float16, variant="fp16")
8
  pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)
 
1
  import gradio as gr
2
  import spaces
3
  import torch
4
+ from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline,AutoPipelineForText2Image, DEISMultistepScheduler
 
5
 
6
  pipe = AutoPipelineForText2Image.from_pretrained('Lykon/AAM_XL_AnimeMix', torch_dtype=torch.float16, variant="fp16")
7
  pipe.scheduler = DEISMultistepScheduler.from_config(pipe.scheduler.config)