Spaces:

eienmojiki
/

Flux-LayerDiffuse

Running on Zero

eienmojiki commited on 4 days ago

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