Irfanuruchi commited on
Commit
fc73351
·
verified ·
1 Parent(s): b20b2ef

Updating README

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -51,14 +51,14 @@ quant_config = BitsAndBytesConfig(
51
  )
52
 
53
  model = AutoModelForCausalLM.from_pretrained(
54
- "IrfanUruchi/Llama-3.1-8B-Computer-Networks-LLM",
55
  quantization_config=quant_config,
56
  torch_dtype=torch.float16,
57
  device_map="auto",
58
  trust_remote_code=True
59
  )
60
 
61
- tokenizer = AutoTokenizer.from_pretrained("IrfanUruchi/Llama-3.1-8B-Computer-Networks-LLM")
62
 
63
  prompt = """You are a network engineering expert. Answer concisely:
64
  Q: What's the difference between TCP and UDP protocols?
 
51
  )
52
 
53
  model = AutoModelForCausalLM.from_pretrained(
54
+ "Irfanuruchi/Llama-3.1-8B-Computer-Networks-LLM",
55
  quantization_config=quant_config,
56
  torch_dtype=torch.float16,
57
  device_map="auto",
58
  trust_remote_code=True
59
  )
60
 
61
+ tokenizer = AutoTokenizer.from_pretrained("Irfanuruchi/Llama-3.1-8B-Computer-Networks-LLM")
62
 
63
  prompt = """You are a network engineering expert. Answer concisely:
64
  Q: What's the difference between TCP and UDP protocols?