Spaces:
Runtime error
Runtime error
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 = [
|