Spaces:

pauri32
/

llm-challenge

Sleeping

pauri32 commited on Aug 24, 2023

Commit

b0fe608

1 Parent(s): 71c4861

Update app/main.py

Files changed (1) hide show

app/main.py CHANGED Viewed

@@ -10,7 +10,7 @@ class InputText(BaseModel):
     text: str
 # "bigscience/bloomz-1b1"
-model_tag = "facebook/opt-125m"
 model = LLM(model_name = model_tag,
             device = "cuda" if torch.cuda.is_available() else "cpu")

     text: str
 # "bigscience/bloomz-1b1"
+model_tag = "bigscience/bloomz-1b1"
 model = LLM(model_name = model_tag,
             device = "cuda" if torch.cuda.is_available() else "cpu")