alfredplpl commited on
Commit
bd32138
1 Parent(s): 416fcc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -1,5 +1,4 @@
1
  from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
2
- from transformers import CLIPFeatureExtractor
3
  import gradio as gr
4
  import torch
5
  from PIL import Image
@@ -14,7 +13,6 @@ model_id = 'Linaqruf/animagine-xl'
14
  auth_token=os.environ["ACCESS_TOKEN"]
15
 
16
  scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler", use_auth_token=auth_token)
17
- feature_extractor = CLIPFeatureExtractor.from_pretrained(model_id, use_auth_token=auth_token)
18
 
19
  pipe_merged = StableDiffusionXLPipeline.from_pretrained(
20
  model_id,
@@ -25,9 +23,7 @@ pipe_i2i_merged = StableDiffusionXLImg2ImgPipeline.from_pretrained(
25
  model_id,
26
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
27
  scheduler=scheduler,
28
- requires_safety_checker=False,
29
- safety_checker=None,
30
- feature_extractor=feature_extractor, use_auth_token=auth_token
31
  )
32
 
33
  pipe=pipe_merged.to("cuda")
 
1
  from diffusers import StableDiffusionXLPipeline, StableDiffusionXLImg2ImgPipeline, EulerAncestralDiscreteScheduler
 
2
  import gradio as gr
3
  import torch
4
  from PIL import Image
 
13
  auth_token=os.environ["ACCESS_TOKEN"]
14
 
15
  scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler", use_auth_token=auth_token)
 
16
 
17
  pipe_merged = StableDiffusionXLPipeline.from_pretrained(
18
  model_id,
 
23
  model_id,
24
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
25
  scheduler=scheduler,
26
+ use_auth_token=auth_token
 
 
27
  )
28
 
29
  pipe=pipe_merged.to("cuda")