stavsap / comfyui-ollama

Apache License 2.0
372 stars 34 forks source link

Allow to keepalive to be hours or seconds [feature request] #64

Open bigcat88 opened 1 week ago

bigcat88 commented 1 week ago

response = client.generate(model=model, prompt=prompt, keep_alive=str(keep_alive) + "m", format=format)

this code can be easy modified:

if keepalive already does not ends with h or s then add m as it currently done.

I can make a PR for that if you wish.

stavsap commented 17 hours ago

a PR for it will be appreciated