Hugging Face's logo

microsoft
/
UserLM-8b

Text Generation
Safetensors
English
llama
userlm
simulation
conversational
Model card Files
xet
Community
7
philippelaban commited on
Commit
e8d3665
·
verified ·
1 Parent(s): 88145e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
32
  import torch
33
 
34
  # Load the model and tokenizer
35
- model_path = "microsoft/userlm-8b"
36
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
37
  model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to("cuda")
38
 
 
32
  import torch
33
 
34
  # Load the model and tokenizer
35
+ model_path = "microsoft/UserLM-8b"
36
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
37
  model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to("cuda")
38