llm-grounded-diffusion

Sleeping

Michael Yang commited on Jul 16, 2023

Commit

de5a7e6

1 Parent(s): e88510d

fix indent

Files changed (1) hide show

baseline.py CHANGED Viewed

@@ -43,7 +43,7 @@ def run(prompt, scheduler_key='dpm_scheduler', bg_seed=1, num_inference_steps=20
     gc.collect()
     torch.cuda.empty_cache()
-     with BytesIO() as buffer:
         np.save(buffer, images[0])
         img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')

     gc.collect()
     torch.cuda.empty_cache()
+    with BytesIO() as buffer:
         np.save(buffer, images[0])
         img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')