mirror of
https://github.com/ollama/ollama-python.git
synced 2026-01-14 06:07:17 +08:00
add example for pulling with progress bar
This commit is contained in:
parent
f00c93eab3
commit
a01b48814a
23
examples/pull-stream-progress/main.py
Normal file
23
examples/pull-stream-progress/main.py
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
# pip install tqdm
|
||||||
|
|
||||||
|
from tqdm import tqdm
|
||||||
|
from ollama import pull
|
||||||
|
|
||||||
|
|
||||||
|
current_digest, bars = '', {}
|
||||||
|
for progress in pull('mistral', stream=True):
|
||||||
|
digest = progress.get('digest', '')
|
||||||
|
if digest != current_digest and current_digest in bars:
|
||||||
|
bars[current_digest].close()
|
||||||
|
|
||||||
|
if not digest:
|
||||||
|
print(progress.get('status'))
|
||||||
|
continue
|
||||||
|
|
||||||
|
if digest not in bars and (total := progress.get('total')):
|
||||||
|
bars[digest] = tqdm(total=total, desc=f'pushing {digest[7:19]}', unit='B', unit_scale=True)
|
||||||
|
|
||||||
|
if completed := progress.get('completed'):
|
||||||
|
bars[digest].update(completed - bars[digest].n)
|
||||||
|
|
||||||
|
current_digest = digest
|
||||||
Loading…
Reference in New Issue
Block a user