Spaces:

ginigen
/

Flux-LayerDiffuse

Runtime error

eienmojiki commited on Feb 8

Commit

69d7a81

verified ·

1 Parent(s): 3a6712a

Update utils/t2i.py

Files changed (1) hide show

utils/t2i.py CHANGED Viewed

@@ -63,5 +63,6 @@ def t2i_gen(
     x = x.clamp(0, 1)
     x = x.permute(0, 2, 3, 1)
     img = Image.fromarray((x*255).float().cpu().numpy().astype(np.uint8)[0])
     return img

     x = x.clamp(0, 1)
     x = x.permute(0, 2, 3, 1)
     img = Image.fromarray((x*255).float().cpu().numpy().astype(np.uint8)[0])
+    torch.cuda.empty_cache()
     return img