Spaces:

DauroCamilo
/

chat-backend-test

Sleeping

DauroCamilo commited on May 27

Commit

7dc77a4

verified ·

1 Parent(s): 0e9b096

GEB-AGI/geb-1.3b

Files changed (1) hide show

main.py CHANGED Viewed

@@ -10,7 +10,7 @@ import threading
 app = FastAPI()
-model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto")

 app = FastAPI()
+model_id = "GEB-AGI/geb-1.3b"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto")