Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
•
cdce2ae
1
Parent(s):
93896d2
Remove inference mode
Browse filesYou do not need the inference mode.
Maybe, it causes the error.
app.py
CHANGED
@@ -70,7 +70,6 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
70 |
|
71 |
|
72 |
@spaces.GPU
|
73 |
-
@torch.inference_mode()
|
74 |
def generate(
|
75 |
prompt: str,
|
76 |
negative_prompt: str = "",
|
|
|
70 |
|
71 |
|
72 |
@spaces.GPU
|
|
|
73 |
def generate(
|
74 |
prompt: str,
|
75 |
negative_prompt: str = "",
|