Spaces:
Runtime error
Runtime error
alfredplpl
commited on
Commit
•
bd32138
1
Parent(s):
416fcc5
Update app.py
Browse files
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 |
-
|
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")
|