Skip to content

Commit

Permalink
Add noqa to pass CI tests
Browse files Browse the repository at this point in the history
  • Loading branch information
plaguss committed Dec 4, 2024
1 parent 92a44bd commit 750f8d6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/distilabel/models/llms/vllm.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,7 +223,7 @@ def unload(self) -> None:
super().unload()

def _cleanup_vllm_model(self) -> None:
import torch
import torch # noqa
from vllm.distributed.parallel_state import (
destroy_distributed_environment,
destroy_model_parallel,
Expand Down

0 comments on commit 750f8d6

Please sign in to comment.