Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
1b1b7c9
1
Parent(s):
e4c2f62
spacesgpu
Browse files
app.py
CHANGED
|
@@ -83,6 +83,7 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict]:
|
|
| 83 |
def get_seed(randomize_seed: bool, seed: int) -> int:
|
| 84 |
return np.random.randint(0, MAX_SEED) if randomize_seed else seed
|
| 85 |
|
|
|
|
| 86 |
def generate_flux_image(
|
| 87 |
prompt: str,
|
| 88 |
seed: int,
|
|
|
|
| 83 |
def get_seed(randomize_seed: bool, seed: int) -> int:
|
| 84 |
return np.random.randint(0, MAX_SEED) if randomize_seed else seed
|
| 85 |
|
| 86 |
+
@spaces.GPU
|
| 87 |
def generate_flux_image(
|
| 88 |
prompt: str,
|
| 89 |
seed: int,
|