try add model_init_args & model_init_kwargs

[yangtianyu92]

def _setup_redis_worker_and_runforever(model_class, batch_size, max_latency, gpu_id, redis_broker, prefix=''):
redis_worker = RedisWorker(model_class, batch_size, max_latency, redis_broker=redis_broker, prefix=prefix)
redis_worker.run_forever(gpu_id)

def run_redis_workers_forever(model_class, batch_size, max_latency=0.1,
worker_num=1, cuda_devices=None, redis_broker="localhost:6379",
prefix='', model_init_args=None, model_init_kwargs=None):
procs = []
for i in range(worker_num):
if cuda_devices is not None:
gpu_id = cuda_devices[i % len(cuda_devices)]
else:
gpu_id = None
args = [model_class, batch_size, max_latency, gpu_id, redis_broker, prefix]
p = mp.Process(target=_setup_redis_worker_and_runforever, args=args, name="stream_worker", daemon=True)
p.start()
procs.append(p)

for p in procs:
    p.join()

model_init_args & model_init_kwargs 这俩参数没往里面传， init_model这个方法读不到参数，得改一下才行。。。所以作者你觉得这是问题还不是个问题？我比较菜，就不乱改了。

[yangtianyu92]

怪了，我是从pip 下载的。。。为什么会不一样呢。。。

[yangtianyu92]

我重新确定了下，github上的代码是可以直接传参数的，我也是这么改的。但是pip安装的确实跟github是不一样的。求作者重新上传下