Spaces:

eienmojiki
/

Flux-LayerDiffuse

Running on Zero

eienmojiki commited on 16 days ago

Commit

e33d4fc

verified ·

1 Parent(s): f7fc0b2

Update utils/t2i.py

Files changed (1) hide show

utils/t2i.py CHANGED Viewed

@@ -33,7 +33,7 @@ trans_vae = TransparentVAE(t2i_pipe.vae, t2i_pipe.vae.dtype)
 trans_vae.load_state_dict(torch.load("./models/TransparentVAE.pth"), strict=False)
 trans_vae.to(device)
-@spaces.GPU(duration=75)
 def t2i_gen(
     prompt: str,
     # negative_prompt: str = None,

 trans_vae.load_state_dict(torch.load("./models/TransparentVAE.pth"), strict=False)
 trans_vae.to(device)
+@spaces.GPU(duration=120)
 def t2i_gen(
     prompt: str,
     # negative_prompt: str = None,