alfredplpl commited on
Commit
cdce2ae
1 Parent(s): 93896d2

Remove inference mode

Browse files

You do not need the inference mode.
Maybe, it causes the error.

Files changed (1) hide show
  1. app.py +0 -1
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 = "",