dnhkng commited on
Commit
de09a79
1 Parent(s): 02dd8cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -15,7 +15,7 @@ import torch
15
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
16
 
17
  torch.random.manual_seed(0)
18
- model_id = "microsoft/Phi-3-medium-4k-instruct"
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_id,
21
  device_map="cuda",
 
15
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
16
 
17
  torch.random.manual_seed(0)
18
+ model_id = "dnhkng/Medium"
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_id,
21
  device_map="cuda",