pva22
commited on
Commit
·
6385f26
1
Parent(s):
f4dfa33
zero gpu usage activate
Browse files
app.py
CHANGED
|
@@ -72,7 +72,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
| 72 |
MAX_IMAGE_SIZE = 1024
|
| 73 |
|
| 74 |
|
| 75 |
-
|
| 76 |
def infer(
|
| 77 |
prompt,
|
| 78 |
negative_prompt,
|
|
@@ -155,4 +155,4 @@ with gr.Blocks(css=css) as demo:
|
|
| 155 |
result = gr.Image(label="Result", show_label=False)
|
| 156 |
|
| 157 |
if __name__ == "__main__":
|
| 158 |
-
demo.launch(
|
|
|
|
| 72 |
MAX_IMAGE_SIZE = 1024
|
| 73 |
|
| 74 |
|
| 75 |
+
@spaces.GPU #[uncomment to use ZeroGPU]
|
| 76 |
def infer(
|
| 77 |
prompt,
|
| 78 |
negative_prompt,
|
|
|
|
| 155 |
result = gr.Image(label="Result", show_label=False)
|
| 156 |
|
| 157 |
if __name__ == "__main__":
|
| 158 |
+
demo.launch()
|