Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|