Spaces:
Running
on
Zero
Running
on
Zero
hatmanstack
commited on
Commit
·
d5ee157
1
Parent(s):
39b6987
rework imports
Browse files- app.py +2 -2
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -3,7 +3,7 @@ sys.path.append('./')
|
|
| 3 |
|
| 4 |
import torch
|
| 5 |
import random
|
| 6 |
-
|
| 7 |
import gradio as gr
|
| 8 |
|
| 9 |
from diffusers import AutoPipelineForText2Image
|
|
@@ -21,7 +21,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
| 21 |
pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=dtype).to(device)
|
| 22 |
pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
|
| 23 |
|
| 24 |
-
|
| 25 |
def create_image(image_pil,
|
| 26 |
prompt,
|
| 27 |
n_prompt,
|
|
|
|
| 3 |
|
| 4 |
import torch
|
| 5 |
import random
|
| 6 |
+
import spaces
|
| 7 |
import gradio as gr
|
| 8 |
|
| 9 |
from diffusers import AutoPipelineForText2Image
|
|
|
|
| 21 |
pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=dtype).to(device)
|
| 22 |
pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
|
| 23 |
|
| 24 |
+
@spaces.GPU(enable_queue=True)
|
| 25 |
def create_image(image_pil,
|
| 26 |
prompt,
|
| 27 |
n_prompt,
|
requirements.txt
CHANGED
|
@@ -3,5 +3,6 @@ torch>=2.0.0
|
|
| 3 |
transformers>=4.37.1
|
| 4 |
spaces>=0.19.4
|
| 5 |
huggingface-hub>=0.20.2
|
|
|
|
| 6 |
gradio
|
| 7 |
accelerate
|
|
|
|
| 3 |
transformers>=4.37.1
|
| 4 |
spaces>=0.19.4
|
| 5 |
huggingface-hub>=0.20.2
|
| 6 |
+
spaces>=0.19.4
|
| 7 |
gradio
|
| 8 |
accelerate
|