Spaces:

daniellefranca96
/

cpu_inf

Sleeping

daniellefranca96 commited on Dec 28, 2023

Commit

c2af308

1 Parent(s): 726fee7

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 import requests
 from ctransformers import AutoModelForCausalLM
 llms = {
   "TinyLLama 1b 4_K_M 2048": {

 from pydantic import BaseModel
 import requests
 from ctransformers import AutoModelForCausalLM
+from llama_cpp import Llama
 llms = {
   "TinyLLama 1b 4_K_M 2048": {