Commit
·
267aac4
1
Parent(s):
0c5448c
update
Browse files- 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,
|