smartdigitalnetworks commited on
Commit
49b52fe
·
verified ·
1 Parent(s): 0be4eb4

Update app_diffusers.py

Browse files
Files changed (1) hide show
  1. app_diffusers.py +1 -1
app_diffusers.py CHANGED
@@ -27,7 +27,7 @@ pipe = OvisImagePipeline.from_pretrained(
27
  )
28
  pipe.to(device)
29
 
30
- @spaces.GPU(duration=75)
31
  def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
32
  print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
33
  generator = torch.Generator().manual_seed(seed)
 
27
  )
28
  pipe.to(device)
29
 
30
+ # @spaces.GPU(duration=75)
31
  def generate(prompt, img_height=1024, img_width=1024, seed=42, steps=50, guidance_scale=5.0):
32
  print(f'inference with prompt : {prompt}, size: {img_height}x{img_width}, seed : {seed}, step : {steps}, cfg : {guidance_scale}')
33
  generator = torch.Generator().manual_seed(seed)