Skip to content

GitLab

Commit f67c0062 authored by comfyanonymous's avatar comfyanonymous
Browse files

Use inference_mode instead of no_grad.

parent 4039616c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
execution.py
View file @ f67c0062
...@@ -143,7 +143,7 @@ class PromptExecutor: ...@@ -143,7 +143,7 @@ class PromptExecutor:
else: else:
self.server.client_id = None self.server.client_id = None
with torch.no_grad(): with torch.inference_mode():
for x in prompt: for x in prompt:
recursive_output_delete_if_changed(prompt, self.old_prompt, self.outputs, x) recursive_output_delete_if_changed(prompt, self.old_prompt, self.outputs, x)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment