Skip to content

Commit

Permalink
updates to the updater
Browse files Browse the repository at this point in the history
  • Loading branch information
lunamidori5 committed Jan 29, 2025
1 parent d82c2c5 commit d58e984
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 2 deletions.
2 changes: 1 addition & 1 deletion Cluster-OS
16 changes: 16 additions & 0 deletions Webserver/Programs/Updater/midori_ai_updater.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,15 @@
["midori_ai_updater", "standard-linux-midori-ai-updater"]
]

cluster_check = [
["midori_ai_login", "midori-ai-login"],
["midori_ai_hf_downloader", "midori-ai-hf-downloader"],
["midori_ai_downloader", "midori-ai-downloader"],
["midori_ai_file_manager", "midori-ai-file-manager"],
["midori_ai_uploader", "midori-ai-uploader"],
["midori_ai_updater", "midori-ai-updater"]
]

with open("/etc/os-release", "r") as f:
os_release_data = f.read()

Expand Down Expand Up @@ -87,6 +96,13 @@
spinner.fail(text=f"Error: Program {program[0]} was not installed successfully.")
del program

for program in cluster_check:
if os.path.isfile("/usr/local/bin/" + program[1]):
shutil.move("/usr/local/bin/" + program[0], "/usr/local/bin/" + program[1])
spinner.succeed(text=f"Program {program[1]} renamed successfully.")
else:
spinner.fail(text=f"Error: Program {program[1]} was not renamed successfully.")

# Clean up
os.chdir("..")
os.rmdir(temp_folder_path)
2 changes: 1 addition & 1 deletion Webserver/other_programs/llm_qna/llm_qna.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@

from transformers import pipeline

pipe = pipeline("text-generation", "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B", torch_dtype=torch.bfloat16, device_map="auto")
pipe = pipeline("text-generation", "deepseek-ai/DeepSeek-R1-Distill-Llama-8B", torch_dtype=torch.bfloat16, device_map="auto")

spinner = Halo(text='Loading', spinner='dots', color='green')

Expand Down

0 comments on commit d58e984

Please sign in to comment.