kashif HF Staff commited on
Commit
c933421
·
verified ·
1 Parent(s): 9d803d5

docs: fix dtype, remove trust_remote_code for model, clean up internal comments

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -41,7 +41,7 @@ import torch
41
  repo = "HuggingFaceBio/Carbon-500M"
42
  tok = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
43
  model = AutoModelForCausalLM.from_pretrained(
44
- repo, torch_dtype=torch.bfloat16,
45
  ).cuda().eval()
46
 
47
  prompt = "<dna>ATGCGCTAGCTACGATCGATCGTAGCTAGCTAGCTAGCTACG"
@@ -60,10 +60,10 @@ import torch
60
 
61
  tok = AutoTokenizer.from_pretrained("HuggingFaceBio/Carbon-3B", trust_remote_code=True)
62
  draft = AutoModelForCausalLM.from_pretrained(
63
- "HuggingFaceBio/Carbon-500M", torch_dtype=torch.bfloat16
64
  ).cuda().eval()
65
  target = AutoModelForCausalLM.from_pretrained(
66
- "HuggingFaceBio/Carbon-3B", torch_dtype=torch.bfloat16
67
  ).cuda().eval()
68
 
69
  prompt = "<dna>ATGCGCTAGCTACGATCGATCGTAGCTAGCTAGCTAGCTACG"
 
41
  repo = "HuggingFaceBio/Carbon-500M"
42
  tok = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
43
  model = AutoModelForCausalLM.from_pretrained(
44
+ repo, dtype=torch.bfloat16,
45
  ).cuda().eval()
46
 
47
  prompt = "<dna>ATGCGCTAGCTACGATCGATCGTAGCTAGCTAGCTAGCTACG"
 
60
 
61
  tok = AutoTokenizer.from_pretrained("HuggingFaceBio/Carbon-3B", trust_remote_code=True)
62
  draft = AutoModelForCausalLM.from_pretrained(
63
+ "HuggingFaceBio/Carbon-500M", dtype=torch.bfloat16
64
  ).cuda().eval()
65
  target = AutoModelForCausalLM.from_pretrained(
66
+ "HuggingFaceBio/Carbon-3B", dtype=torch.bfloat16
67
  ).cuda().eval()
68
 
69
  prompt = "<dna>ATGCGCTAGCTACGATCGATCGTAGCTAGCTAGCTAGCTACG"