change rerender lora to work with storyteller-rust

This commit is contained in:
salt
2023-12-12 23:11:45 -08:00
parent 88f8fb7a62
commit c926010fda
4 changed files with 38 additions and 3 deletions
+2 -1
View File
@@ -2,6 +2,7 @@
# CrossAttn precision handling
import os
import sys
import einops
import numpy as np
@@ -290,7 +291,7 @@ class DDIMVSampler(object):
else timesteps.shape[0]
print(f'Running DDIM Sampling with {total_steps} timesteps')
iterator = tqdm(time_range, desc='DDIM Sampler', total=total_steps)
iterator = tqdm(time_range, desc='DDIM Sampler', total=total_steps, file=sys.stdout)
if controller is not None:
controller.set_total_step(total_steps)
if mask is None: