qnguyen3 commited on
Commit
f76e333
1 Parent(s): ee9edb4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ warnings.filterwarnings('ignore')
47
  # set device
48
  torch.set_default_device('cuda') # or 'cpu'
49
 
50
- model_name = 'fne/dolphin-llava-qwen2-72b'
51
 
52
  # create model
53
  model = AutoModelForCausalLM.from_pretrained(
 
47
  # set device
48
  torch.set_default_device('cuda') # or 'cpu'
49
 
50
+ model_name = 'cognitivecomputations/dolphin-vision-72b'
51
 
52
  # create model
53
  model = AutoModelForCausalLM.from_pretrained(