linoyts HF staff commited on
Commit
d02daeb
1 Parent(s): 2353090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ from torch import autocast, inference_mode
11
  import re
12
  import spaces
13
 
14
- @spaces.GPU
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 = [