mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 14:52:20 +00:00
26 lines
604 B
Python
26 lines
604 B
Python
from itertools import chain
|
|
|
|
import psutil
|
|
|
|
|
|
class SweepParameter:
|
|
def __init__(self, name, start, end, step):
|
|
self.name = name
|
|
self.start = start
|
|
self.end = end
|
|
self.step = step
|
|
|
|
def __str__(self):
|
|
return f"{self.name} from {self.start} to {self.end} in steps of {self.step}"
|
|
|
|
def __iter__(self):
|
|
for value in chain(range(self.start, self.end, self.step), [self.end]):
|
|
yield value
|
|
|
|
|
|
def kill(proc_pid):
|
|
process = psutil.Process(proc_pid)
|
|
for proc in process.children(recursive=True):
|
|
proc.kill()
|
|
process.kill()
|