Upload 2 files

Dockerfile

@@ -0,0 +1,12 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY ./pt_merge_model_v3.Q4_K_M.gguf /code/pt_merge_model_v3.Q4_K_M.gguf
+COPY ./main.py /code/main.py
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,25 @@
+from ctransformers import AutoModelForCausalLM
+from fastapi import FastAPI, Form
+from pydantic import BaseModel
+
+#Model loading
+llm = AutoModelForCausalLM.from_pretrained("pt_merge_model_v3.Q4_K_M.gguf",
+model_type='llama',
+max_new_tokens = 512,
+threads = 3,
+)
+   
+
+#Pydantic object
+class validation(BaseModel):
+    prompt: str
+    
+#Fast API
+app = FastAPI()
+
+#Zephyr completion
+@app.post("/llama_on_cpu")
+async def stream(item: validation):
+    system_prompt = 'Below is an instruction that describes a task. Write a response that appropriately completes the request.'
+    prompt =  f"<s>[INST]<<SYS>>\n + {system_prompt} + <</SYS>>\n{item.prompt.strip()}[/INST]"
+    return llm(prompt)