Spaces:
Runtime error
Runtime error
Update text_to_image.py
Browse files- text_to_image.py +0 -2
text_to_image.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
from transformers.agents.tools import Tool
|
| 2 |
from transformers.utils import is_accelerate_available
|
| 3 |
import torch
|
| 4 |
-
import spaces
|
| 5 |
|
| 6 |
|
| 7 |
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
|
@@ -12,7 +11,6 @@ TEXT_TO_IMAGE_DESCRIPTION = (
|
|
| 12 |
"This is a tool that creates an image according to a prompt."
|
| 13 |
)
|
| 14 |
|
| 15 |
-
@spaces.GPU
|
| 16 |
class TextToImageTool(Tool):
|
| 17 |
default_checkpoint = "runwayml/stable-diffusion-v1-5"
|
| 18 |
description = TEXT_TO_IMAGE_DESCRIPTION
|
|
|
|
| 1 |
from transformers.agents.tools import Tool
|
| 2 |
from transformers.utils import is_accelerate_available
|
| 3 |
import torch
|
|
|
|
| 4 |
|
| 5 |
|
| 6 |
from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
|
|
|
|
| 11 |
"This is a tool that creates an image according to a prompt."
|
| 12 |
)
|
| 13 |
|
|
|
|
| 14 |
class TextToImageTool(Tool):
|
| 15 |
default_checkpoint = "runwayml/stable-diffusion-v1-5"
|
| 16 |
description = TEXT_TO_IMAGE_DESCRIPTION
|