Spaces:

THUDM
/

CogView2

Runtime error

hysts HF staff commited on Jul 29, 2022

Commit

e080c62

•

1 Parent(s): 598b16c

Update

Files changed (1) hide show

model.py CHANGED Viewed

@@ -294,6 +294,7 @@ class Model:
             torch.cuda.empty_cache()
             self.srg.dsr.model.to(self.device)
             self.srg.itersr.model.to(self.device)
         res = self.generate_images(seq, txt_len, tokens)
         elapsed = time.perf_counter() - start

             torch.cuda.empty_cache()
             self.srg.dsr.model.to(self.device)
             self.srg.itersr.model.to(self.device)
+        torch.cuda.empty_cache()
         res = self.generate_images(seq, txt_len, tokens)
         elapsed = time.perf_counter() - start