kamran-r123 commited on
Commit
f0a5811
·
verified ·
1 Parent(s): a347da3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -8,8 +8,8 @@ from huggingface_hub import hf_hub_download
8
 
9
 
10
  model_id = "failspy/Meta-Llama-3-8B-Instruct-abliterated-v3-GGUF"
11
- # model_path = hf_hub_download(repo_id=model_id, filename="Meta-Llama-3-8B-Instruct-abliterated-v3_q6.gguf", token=os.environ['HF_TOKEN'])
12
- # model = Llama(model_path=model_path, n_gpu_layers=-1, n_ctx=4096, verbose=False)
13
 
14
  class Item(BaseModel):
15
  prompt: str
@@ -52,4 +52,4 @@ async def generate_text(item: Item):
52
  @app.get("/")
53
  def read_root():
54
 
55
- return {"Hello": os.environ['HF_TOKEN']}
 
8
 
9
 
10
  model_id = "failspy/Meta-Llama-3-8B-Instruct-abliterated-v3-GGUF"
11
+ model_path = hf_hub_download(repo_id=model_id, filename="Meta-Llama-3-8B-Instruct-abliterated-v3_q6.gguf", token=os.environ['HF_TOKEN'])
12
+ model = Llama(model_path=model_path, n_gpu_layers=-1, n_ctx=4096, verbose=False)
13
 
14
  class Item(BaseModel):
15
  prompt: str
 
52
  @app.get("/")
53
  def read_root():
54
 
55
+ return {"Hello": "Worlds"}