Michael Yang
commited on
Commit
•
de5a7e6
1
Parent(s):
e88510d
fix indent
Browse files- baseline.py +1 -1
baseline.py
CHANGED
|
@@ -43,7 +43,7 @@ def run(prompt, scheduler_key='dpm_scheduler', bg_seed=1, num_inference_steps=20
|
|
| 43 |
gc.collect()
|
| 44 |
torch.cuda.empty_cache()
|
| 45 |
|
| 46 |
-
|
| 47 |
np.save(buffer, images[0])
|
| 48 |
img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
|
| 49 |
|
|
|
|
| 43 |
gc.collect()
|
| 44 |
torch.cuda.empty_cache()
|
| 45 |
|
| 46 |
+
with BytesIO() as buffer:
|
| 47 |
np.save(buffer, images[0])
|
| 48 |
img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
|
| 49 |
|