cocktailpeanut commited on
Commit
3961dc5
·
1 Parent(s): f5e8315
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  import cv2
3
  import math
4
- import spaces
5
  import torch
6
  import random
7
  import numpy as np
@@ -183,7 +183,7 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str
183
  p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
184
  return p.replace("{prompt}", positive), n + ' ' + negative
185
 
186
- @spaces.GPU
187
  def generate_image(face_image, pose_image, prompt, negative_prompt, style_name, enhance_face_region, num_steps, identitynet_strength_ratio, adapter_strength_ratio, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
188
 
189
  if face_image is None:
 
1
  import os
2
  import cv2
3
  import math
4
+ #import spaces
5
  import torch
6
  import random
7
  import numpy as np
 
183
  p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
184
  return p.replace("{prompt}", positive), n + ' ' + negative
185
 
186
+ #@spaces.GPU
187
  def generate_image(face_image, pose_image, prompt, negative_prompt, style_name, enhance_face_region, num_steps, identitynet_strength_ratio, adapter_strength_ratio, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
188
 
189
  if face_image is None: