xywwww commited on
Commit
7e98b1f
·
verified ·
1 Parent(s): e6efdca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ model.load_state_dict(load_state_dict(ckpt), strict=False)
30
  model = model.cuda()
31
  ddim_sampler = DDIMSampler(model)
32
 
33
- +@spaces.GPU
34
  def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
35
  with torch.no_grad():
36
  img = resize_image(HWC3(input_image), image_resolution)
 
30
  model = model.cuda()
31
  ddim_sampler = DDIMSampler(model)
32
 
33
+
34
  def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
35
  with torch.no_grad():
36
  img = resize_image(HWC3(input_image), image_resolution)