SSTalha commited on
Commit
e9e8565
·
verified ·
1 Parent(s): c2aa038

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -39,7 +39,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
39
 
40
  torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
41
 
42
- Model = 'SSTalha/Fashion_Phi-1_5'
43
  model = AutoModelForCausalLM.from_pretrained(Model, torch_dtype="auto", trust_remote_code=True).to(torch_device)
44
  tokenizer = AutoTokenizer.from_pretrained(Model, trust_remote_code=True)
45
 
 
39
 
40
  torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
41
 
42
+ Model = 'SSTalha/Fashion_PHI_1-5'
43
  model = AutoModelForCausalLM.from_pretrained(Model, torch_dtype="auto", trust_remote_code=True).to(torch_device)
44
  tokenizer = AutoTokenizer.from_pretrained(Model, trust_remote_code=True)
45