X-prime commited on
Commit
a1f104e
·
verified ·
1 Parent(s): 12bfa6b

update :heavy_check_mark:

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ import gradio as gr
67
  from huggingface_hub import InferenceClient
68
  from transformers import AutoTokenizer, AutoModelForCausalLM
69
 
70
- model_name = "openbiolabs/openbiollm-70b"
71
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
72
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", load_in_8bit=True)
73
 
 
67
  from huggingface_hub import InferenceClient
68
  from transformers import AutoTokenizer, AutoModelForCausalLM
69
 
70
+ model_name = "aaditya/Llama3-OpenBioLLM-70B"
71
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
72
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", load_in_8bit=True)
73