REILX commited on
Commit
fb75b16
·
verified ·
1 Parent(s): b71556b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ import torch
42
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
43
 
44
  torch.random.manual_seed(0)
45
- model_id = "/home/intelisu/models/weight/phi3/Phi-3-medium-128k-instruct/"
46
  model = AutoModelForCausalLM.from_pretrained(
47
  model_id,
48
  device_map="cuda",
 
42
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
43
 
44
  torch.random.manual_seed(0)
45
+ model_id = "/home/models/phi3/Phi-3-medium-128k-instruct/"
46
  model = AutoModelForCausalLM.from_pretrained(
47
  model_id,
48
  device_map="cuda",