Spaces:
Running
on
Zero
Running
on
Zero
Prime Cai
commited on
Commit
·
fd5b7c3
1
Parent(s):
a1ef85e
fix dynamic gpu
Browse files
app.py
CHANGED
|
@@ -38,7 +38,14 @@ pipe.load_lora_weights(
|
|
| 38 |
pipe.to(device, dtype=dtype)
|
| 39 |
|
| 40 |
@spaces.GPU
|
| 41 |
-
def generate_image(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
w, h, min_size = image.size[0], image.size[1], min(image.size)
|
| 43 |
image = image.crop(
|
| 44 |
((w - min_size) // 2, (h - min_size) // 2, (w + min_size) // 2, (h + min_size) // 2)
|
|
|
|
| 38 |
pipe.to(device, dtype=dtype)
|
| 39 |
|
| 40 |
@spaces.GPU
|
| 41 |
+
def generate_image(
|
| 42 |
+
image: Image.Image,
|
| 43 |
+
text: str,
|
| 44 |
+
gemini_prompt: bool = True,
|
| 45 |
+
guidance: float = 3.5,
|
| 46 |
+
i_guidance: float = 1.0,
|
| 47 |
+
t_guidance: float = 1.0
|
| 48 |
+
):
|
| 49 |
w, h, min_size = image.size[0], image.size[1], min(image.size)
|
| 50 |
image = image.crop(
|
| 51 |
((w - min_size) // 2, (h - min_size) // 2, (w + min_size) // 2, (h + min_size) // 2)
|