WaltonFuture commited on
Commit
e0f22b9
·
verified ·
1 Parent(s): d07bd88

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
42
  import torch
43
 
44
  device = "cuda" # the device to load the model onto
45
- model_path = 'WaltonFuture/Diabetica-7B'
46
 
47
  model = AutoModelForCausalLM.from_pretrained(
48
  model_path,
 
42
  import torch
43
 
44
  device = "cuda" # the device to load the model onto
45
+ model_path = 'WaltonFuture/Diabetica-1.5B'
46
 
47
  model = AutoModelForCausalLM.from_pretrained(
48
  model_path,