This commit is contained in:
lllyasviel
2023-10-12 04:23:10 -07:00
committed by GitHub
parent 4c867c1b8b
commit e61aac34ca
147 changed files with 523 additions and 642 deletions
+3 -3
View File
@@ -20,7 +20,7 @@ def worker():
import modules.flags as flags
import modules.path
import modules.patch
import comfy.model_management
import fcbh.model_management
import fooocus_extras.preprocessors as preprocessors
import modules.inpaint_worker as inpaint_worker
import modules.advanced_parameters as advanced_parameters
@@ -483,7 +483,7 @@ def worker():
outputs.append(['preview', (13, 'Moving model to GPU ...', None)])
execution_start_time = time.perf_counter()
comfy.model_management.load_models_gpu([pipeline.final_unet])
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')
@@ -558,7 +558,7 @@ def worker():
log(x, d, single_line_number=3)
results += imgs
except comfy.model_management.InterruptProcessingException as e:
except fcbh.model_management.InterruptProcessingException as e:
if shared.last_stop == 'skip':
print('User skipped')
continue