Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ from torch import autocast, inference_mode
|
|
| 11 |
import re
|
| 12 |
import spaces
|
| 13 |
|
| 14 |
-
|
| 15 |
def randomize_seed_fn(seed, randomize_seed):
|
| 16 |
if randomize_seed:
|
| 17 |
seed = random.randint(0, np.iinfo(np.int32).max)
|
|
@@ -59,7 +59,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
| 59 |
sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
|
| 60 |
sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
|
| 61 |
|
| 62 |
-
|
| 63 |
|
| 64 |
def get_example():
|
| 65 |
case = [
|
|
|
|
| 11 |
import re
|
| 12 |
import spaces
|
| 13 |
|
| 14 |
+
|
| 15 |
def randomize_seed_fn(seed, randomize_seed):
|
| 16 |
if randomize_seed:
|
| 17 |
seed = random.randint(0, np.iinfo(np.int32).max)
|
|
|
|
| 59 |
sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
|
| 60 |
sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
|
| 61 |
|
| 62 |
+
@spaces.GPU
|
| 63 |
|
| 64 |
def get_example():
|
| 65 |
case = [
|