minor bug fix praisonai train

Dockerfile

@@ -1,6 +1,6 @@
 FROM python:3.11-slim
 WORKDIR /app
 COPY . .
-RUN pip install flask praisonai==0.0.60 gunicorn markdown
+RUN pip install flask praisonai==0.0.61 gunicorn markdown
 EXPOSE 8080
 CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]

config.yaml

@@ -1,10 +1,10 @@
-ollama_save: "false"
-huggingface_save: "false"
-train: "false"
+ollama_save: "true"
+huggingface_save: "true"
+train: "true"
 
 model_name: "unsloth/Meta-Llama-3.1-8B-Instruct-bnb-4bit"
-hf_model_name: "mervinpraison/llama-3.1-tamilan-8B-test"
-ollama_model: "mervinpraison/llama3.1-tamilan-test"
+hf_model_name: "mervinpraison/llama-3.1-instruct"
+ollama_model: "mervinpraison/llama3.1-instruct"
 model_parameters: "8b"
 
 dataset:

docs/api/praisonai/deploy.html

@@ -110,7 +110,7 @@ Loads environment variables from .env file or system and sets them.</p>
             file.write(&#34;FROM python:3.11-slim\n&#34;)
             file.write(&#34;WORKDIR /app\n&#34;)
             file.write(&#34;COPY . .\n&#34;)
-            file.write(&#34;RUN pip install flask praisonai==0.0.60 gunicorn markdown\n&#34;)
+            file.write(&#34;RUN pip install flask praisonai==0.0.61 gunicorn markdown\n&#34;)
             file.write(&#34;EXPOSE 8080\n&#34;)
             file.write(&#39;CMD [&#34;gunicorn&#34;, &#34;-b&#34;, &#34;0.0.0.0:8080&#34;, &#34;api:app&#34;]\n&#39;)
             

poetry.lock

@@ -808,7 +808,7 @@ fastapi = ">=0.110.1,<0.111.0"
 filetype = ">=1.2.0,<2.0.0"
 httpx = ">=0.23.0"
 lazify = ">=0.4.0,<0.5.0"
-literalai = "0.0.607"
+literalai = "0.0.617"
 nest-asyncio = ">=1.5.6,<2.0.0"
 numpy = {version = ">=1.26,<2.0", markers = "python_version >= \"3.9\""}
 packaging = ">=23.1,<24.0"
@@ -3702,12 +3702,12 @@ proxy = ["PyJWT (>=2.8.0,<3.0.0)", "apscheduler (>=3.10.4,<4.0.0)", "backoff", "
 
 [[package]]
 name = "literalai"
-version = "0.0.607"
+version = "0.0.617"
 description = "An SDK for observability in Python applications"
 optional = false
 python-versions = "*"
 files = [
-    {file = "literalai-0.0.607.tar.gz", hash = "sha256:783c495d9fb2ae9f84e5877ecb34587f6a6ac0224d9f0936336dbc8b6765b30d"},
+    {file = "literalai-0.0.617.tar.gz", hash = "sha256:783c495d9fb2ae9f84e5877ecb34587f6a6ac0224d9f0936336dbc8b6765b30d"},
 ]
 
 [package.dependencies]

praisonai.rb

@@ -3,7 +3,7 @@ class Praisonai < Formula
   
     desc "AI tools for various AI applications"
     homepage "https://github.com/MervinPraison/PraisonAI"
-    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/0.0.60.tar.gz"
+    url "https://github.com/MervinPraison/PraisonAI/archive/refs/tags/0.0.61.tar.gz"
     sha256 "1828fb9227d10f991522c3f24f061943a254b667196b40b1a3e4a54a8d30ce32"  # Replace with actual SHA256 checksum
     license "MIT"
   

praisonai/deploy.py

@@ -56,7 +56,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==0.0.60 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==0.0.61 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

praisonai/train.py

@@ -245,7 +245,7 @@ PARAMETER stop "<|reserved_special_token_"
             self.save_model_merged()
 
         if self.config.get("huggingface_save_gguf", "true").lower() == "true":
-            self.model, self.tokenizer = self.load_model()
+            # self.model, self.tokenizer = self.load_model()
             self.push_model_gguf()
             
         # if self.config.get("save_gguf", "true").lower() == "true": ## TODO

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "PraisonAI"
-version = "0.0.60"
+version = "0.0.61"
 description = "PraisonAI application combines AutoGen and CrewAI or similar frameworks into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customization, and efficient human-agent collaboration."
 authors = ["Mervin Praison"]
 license = ""