Update README.md
Browse files
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/
|
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 |
|