Spaces:

nroggendorff
/

dolphin

Paused

nroggendorff commited on May 26, 2024

Commit

3792f85

verified ·

1 Parent(s): 757022d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ def predict(message, history):
     torch.set_default_device("cuda")
     tokenizer = AutoTokenizer.from_pretrained(
-        "cognitivecomputations/dolphin-2.8-mistral-7b-v02",
         trust_remote_code=True
     )
     model = AutoModelForCausalLM.from_pretrained(

     torch.set_default_device("cuda")
     tokenizer = AutoTokenizer.from_pretrained(
+        "cognitivecomputations/dolphin-2.9.1-mixtral-1x22b",
         trust_remote_code=True
     )
     model = AutoModelForCausalLM.from_pretrained(