Update backend_pytorch.py | Fix lock usage

mlcommons · Dec 8, 2024 · 20dfb0d · 20dfb0d
1 parent dd9be96
commit 20dfb0d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/text_to_image/backend_pytorch.py b/text_to_image/backend_pytorch.py
@@ -387,7 +387,7 @@ def predict(self, inputs):
                     pooled_prompt_embeds,
                     negative_pooled_prompt_embeds,
                 ) = self.prepare_inputs(inputs, i)
-                with lock:
+                with self.lock:
                     generated = self.pipe(
                         prompt_embeds=prompt_embeds,
                         negative_prompt_embeds=negative_prompt_embeds,