RealVis_v5.0_BF16_IP_B

Sleeping

1inkusFace commited on Jan 21

Commit

5116401

verified ·

1 Parent(s): c919ac4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -462,7 +462,7 @@ def generate_30(
         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
-        torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
@@ -591,7 +591,7 @@ def generate_60(
         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
-        torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
@@ -720,7 +720,7 @@ def generate_90(
         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
-        torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")

         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
+        #torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
+        #torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")
         del processor5
         gc.collect()
         #torch.cuda.empty_cache()
+        #torch.cuda.reset_peak_memory_stats()
         #expanded = expand_prompt(captions)
         new_prompt = prompt + ' ' + captions
         print("-- ------------ --")