EnjoyCodeX commited on
Commit
ba96a26
1 Parent(s): b436982

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -64,16 +64,16 @@ Use the code below to get started with the model.
64
  <div class="relative group repo-copy-code"><pre><code class="language-python">
65
  def init_model():
66
  model = AutoModelForCausalLM.from_pretrained(
67
- "./MedLang13B",
68
  torch_dtype=torch.float16,
69
  device_map="auto",
70
  trust_remote_code=True
71
  )
72
  model.generation_config = GenerationConfig.from_pretrained(
73
- "./MedLang13B",
74
  )
75
  tokenizer = AutoTokenizer.from_pretrained(
76
- "./MedLang13B",
77
  use_fast=False,
78
  trust_remote_code=True
79
  )
 
64
  <div class="relative group repo-copy-code"><pre><code class="language-python">
65
  def init_model():
66
  model = AutoModelForCausalLM.from_pretrained(
67
+ "EnjoyCodeX/MedLang-13B/MedLang-13B",
68
  torch_dtype=torch.float16,
69
  device_map="auto",
70
  trust_remote_code=True
71
  )
72
  model.generation_config = GenerationConfig.from_pretrained(
73
+ "EnjoyCodeX/MedLang-13B/MedLang-13B",
74
  )
75
  tokenizer = AutoTokenizer.from_pretrained(
76
+ "EnjoyCodeX/MedLang-13B/MedLang-13B",
77
  use_fast=False,
78
  trust_remote_code=True
79
  )