Spaces:

Ketengan-Diffusion-Lab
/

Dolphin-Inference

Build error

Ketengan-Diffusion-Lab commited on Sep 15

Commit

cda5511

•

1 Parent(s): b1bbde3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ warnings.filterwarnings('ignore')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
-model_name = 'cognitivecomputations/dolphin-vision-7b'
 # create model and load it to the specified device
 model = AutoModelForCausalLM.from_pretrained(

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"Using device: {device}")
+model_name = 'cognitivecomputations/dolphin-vision-72b'
 # create model and load it to the specified device
 model = AutoModelForCausalLM.from_pretrained(