EpistemeAI
/

Iceball-Phi-3.5-mini-instruct-shareGPT-v1.0

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

legolasyiu commited on Aug 27

Commit

368c114

•

1 Parent(s): b8d2592

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -73,12 +73,12 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
-    "microsoft/Phi-3.5-mini-instruct",
     device_map="cuda",
     torch_dtype="auto",
     trust_remote_code=True,
 )
-tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3.5-mini-instruct")
 messages = [
     {"role": "system", "content": "You are a helpful AI assistant."},
     {"role": "user", "content": "Can you provide ways to eat combinations of bananas and dragonfruits?"},

 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 torch.random.manual_seed(0)
 model = AutoModelForCausalLM.from_pretrained(
+    "EpistemeAI/Iceball-Phi-3.5-mini-instruct-shareGPT-v1.0",
     device_map="cuda",
     torch_dtype="auto",
     trust_remote_code=True,
 )
+tokenizer = AutoTokenizer.from_pretrained("EpistemeAI/Iceball-Phi-3.5-mini-instruct-shareGPT-v1.0")
 messages = [
     {"role": "system", "content": "You are a helpful AI assistant."},
     {"role": "user", "content": "Can you provide ways to eat combinations of bananas and dragonfruits?"},