ffreemt commited on
Commit
d89e980
1 Parent(s): fd0376c

Update wget aria2 in packages.txt

Browse files
Files changed (2) hide show
  1. app.py +11 -2
  2. packages.txt +1 -0
app.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from __future__ import annotations
2
 
3
  import os
@@ -15,6 +17,9 @@ from huggingface_hub import hf_hub_download, hf_hub_url # snapshot_download,
15
  from loguru import logger
16
  from python_run_cmd import run_cmd
17
 
 
 
 
18
  os.environ["TZ"] = "Asia/Shanghai"
19
  try:
20
  time.tzset() # type: ignore
@@ -28,7 +33,9 @@ filename = "openbuddy-mistral-7b-v13.Q4_K_M.gguf" # 4.39G
28
 
29
  model_ready = True
30
  logger.debug("Start dl")
 
31
  # try to download 5 times:
 
32
  for idx in range(5):
33
  logger.debug(f"attempt {idx + 1}")
34
  try:
@@ -53,7 +60,7 @@ if not model_ready: # try aria2c
53
  url = hf_hub_url(
54
  repo_id,
55
  filename,
56
- # revision="ggmlv3",
57
  )
58
  logger.debug(f"{url=}")
59
  ret = run_cmd(f"wget -c {url}", mute_stdout=False)
@@ -62,7 +69,7 @@ if not model_ready: # try aria2c
62
  model_path = f"./{filename}"
63
 
64
  # both successful
65
- if not ret.returncode and not ret1.returncode:
66
  model_ready = True
67
 
68
  # Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
@@ -138,6 +145,8 @@ css = ".generating {visibility: hidden}"
138
 
139
  # Based on the gradio theming guide and borrowed from https://huggingface.co/spaces/shivi/dolly-v2-demo
140
  class SeafoamCustom(Base):
 
 
141
  def __init__(
142
  self,
143
  *,
 
1
+ """Run."""
2
+ # pylint: disable=invalid-name,line-too-long,broad-except,missing-function-docstring
3
  from __future__ import annotations
4
 
5
  import os
 
17
  from loguru import logger
18
  from python_run_cmd import run_cmd
19
 
20
+ ret = run_cmd("which aria2c", mute_stdout=False)
21
+ logger.debug(ret)
22
+
23
  os.environ["TZ"] = "Asia/Shanghai"
24
  try:
25
  time.tzset() # type: ignore
 
33
 
34
  model_ready = True
35
  logger.debug("Start dl")
36
+
37
  # try to download 5 times:
38
+ model_path = f"./{filename}"
39
  for idx in range(5):
40
  logger.debug(f"attempt {idx + 1}")
41
  try:
 
60
  url = hf_hub_url(
61
  repo_id,
62
  filename,
63
+ # revision="main",
64
  )
65
  logger.debug(f"{url=}")
66
  ret = run_cmd(f"wget -c {url}", mute_stdout=False)
 
69
  model_path = f"./{filename}"
70
 
71
  # both successful
72
+ if not ret.returncode:
73
  model_ready = True
74
 
75
  # Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
 
145
 
146
  # Based on the gradio theming guide and borrowed from https://huggingface.co/spaces/shivi/dolly-v2-demo
147
  class SeafoamCustom(Base):
148
+ """Define."""
149
+
150
  def __init__(
151
  self,
152
  *,
packages.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ aria2