Nymbo commited on
Commit
09b3d46
·
verified ·
1 Parent(s): 83e4367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  import numpy as np
3
  import random
4
 
5
- import spaces
6
  import torch
7
 
8
  from diffusers import FluxPriorReduxPipeline, FluxPipeline
@@ -23,7 +23,7 @@ pipe = FluxPipeline.from_pretrained(
23
  torch_dtype=torch.bfloat16
24
  ).to("cuda")
25
 
26
- @spaces.GPU
27
  def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
28
  if randomize_seed:
29
  seed = random.randint(0, MAX_SEED)
 
2
  import numpy as np
3
  import random
4
 
5
+ # import spaces
6
  import torch
7
 
8
  from diffusers import FluxPriorReduxPipeline, FluxPipeline
 
23
  torch_dtype=torch.bfloat16
24
  ).to("cuda")
25
 
26
+ # @spaces.GPU
27
  def infer(control_image, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
28
  if randomize_seed:
29
  seed = random.randint(0, MAX_SEED)