Spaces:
Runtime error
Runtime error
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):
|