This commit is contained in:
lvmin
2023-10-13 04:53:02 -07:00
parent c4118d2418
commit 10e2db67d6
2 changed files with 14 additions and 6 deletions
-6
View File
@@ -482,12 +482,6 @@ def worker():
print(f'Preparation time: {preparation_time:.2f} seconds')
outputs.append(['preview', (13, 'Moving model to GPU ...', None)])
execution_start_time = time.perf_counter()
fcbh.model_management.load_models_gpu([pipeline.final_unet])
moving_time = time.perf_counter() - execution_start_time
print(f'Moving model to GPU: {moving_time:.2f} seconds')
outputs.append(['preview', (13, 'Starting tasks ...', None)])
def callback(step, x0, x, total_steps, y):
done_steps = current_task_id * steps + step