Uhhy commited on
Commit
5fd0205
1 Parent(s): 1ee8e15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -97,7 +97,6 @@ MAX_SEED = np.iinfo(np.int32).max
97
  MAX_IMAGE_SIZE = 2048
98
 
99
  @spaces.GPU()
100
- @GPU(duration=1)
101
  def generate_model_response(model, inputs):
102
  try:
103
  response = model(inputs)
@@ -107,7 +106,6 @@ def generate_model_response(model, inputs):
107
  return ""
108
 
109
  @spaces.GPU()
110
- @GPU(duration=1)
111
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4):
112
  if randomize_seed:
113
  seed = random.randint(0, MAX_SEED)
 
97
  MAX_IMAGE_SIZE = 2048
98
 
99
  @spaces.GPU()
 
100
  def generate_model_response(model, inputs):
101
  try:
102
  response = model(inputs)
 
106
  return ""
107
 
108
  @spaces.GPU()
 
109
  def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_inference_steps=4):
110
  if randomize_seed:
111
  seed = random.randint(0, MAX_SEED)