Spaces:
Runtime error
Runtime error
Update gradio_demo.py
Browse files- gradio_demo.py +1 -0
gradio_demo.py
CHANGED
|
@@ -277,6 +277,7 @@ i2i_pipe = StableDiffusionImg2ImgPipeline(
|
|
| 277 |
)
|
| 278 |
|
| 279 |
|
|
|
|
| 280 |
@torch.inference_mode()
|
| 281 |
def encode_prompt_inner(txt: str):
|
| 282 |
max_length = tokenizer.model_max_length
|
|
|
|
| 277 |
)
|
| 278 |
|
| 279 |
|
| 280 |
+
@spaces.GPU(duration=60)
|
| 281 |
@torch.inference_mode()
|
| 282 |
def encode_prompt_inner(txt: str):
|
| 283 |
max_length = tokenizer.model_max_length
|