Hugging Face's logo

Spaces:
LLKIM
/
project
Runtime error

App Files Community
1
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):