Spaces:
Sleeping
Sleeping
Update space
Browse files- model/animagine_xl.py +1 -1
model/animagine_xl.py
CHANGED
@@ -33,7 +33,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
|
|
33 |
variant="fp16"
|
34 |
)
|
35 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
36 |
-
pipe = pipe.to("
|
37 |
|
38 |
# (2) ์ด๋ฏธ์ง ์์ฑ ํจ์
|
39 |
def generate_animagine_xl(prompt: str):
|
|
|
33 |
variant="fp16"
|
34 |
)
|
35 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
36 |
+
pipe = pipe.to("cpu") # ๋๋ "cpu"๋ก ๋ณ๊ฒฝ ๊ฐ๋ฅ
|
37 |
|
38 |
# (2) ์ด๋ฏธ์ง ์์ฑ ํจ์
|
39 |
def generate_animagine_xl(prompt: str):
|