rnlduatm commited on
Commit
96533d3
ยท
1 Parent(s): dcb02b2

Update space

Browse files
Files changed (1) hide show
  1. 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("cuda") # ๋˜๋Š” "cpu"๋กœ ๋ณ€๊ฒฝ ๊ฐ€๋Šฅ
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):