cocktailpeanut commited on
Commit
267aac4
·
1 Parent(s): 0c5448c
Files changed (1) hide show
  1. injection_main_HF.py +2 -2
injection_main_HF.py CHANGED
@@ -516,7 +516,7 @@ if __name__ == "__main__":
516
  if mode == "app":
517
  # gradio
518
  import gradio as gr
519
- import spaces
520
 
521
  # Load a pipeline
522
  pipe = StableDiffusionPipeline.from_pretrained(
@@ -526,7 +526,7 @@ if __name__ == "__main__":
526
  # Set up a DDIM scheduler
527
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
528
 
529
- @spaces.GPU()
530
  def style_transfer_app(
531
  prompt,
532
  image,
 
516
  if mode == "app":
517
  # gradio
518
  import gradio as gr
519
+ # import spaces
520
 
521
  # Load a pipeline
522
  pipe = StableDiffusionPipeline.from_pretrained(
 
526
  # Set up a DDIM scheduler
527
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
528
 
529
+ # @spaces.GPU()
530
  def style_transfer_app(
531
  prompt,
532
  image,