Hugging Face's logo

Spaces:
rashmi
/
h2oai-predict-llm
Sleeping

App Files Community
rashmi commited on
Commit
b5fc670
·
1 Parent(s): 31ee432

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ theme = gr.themes.Monochrome(
51
  ### Load the model
52
  class CFG:
53
  num_workers = os.cpu_count()
54
- llm_backbone = "save_pretrained_model/zephyr-7b-beta"
55
  tokenizer_path = "HuggingFaceH4/zephyr-7b-beta"
56
  tokenizer = AutoTokenizer.from_pretrained(
57
  tokenizer_path, add_prefix_space=False, use_fast=True, trust_remote_code=True, add_eos_token=True
 
51
  ### Load the model
52
  class CFG:
53
  num_workers = os.cpu_count()
54
+ llm_backbone = "HuggingFaceH4/zephyr-7b-beta" #"save_pretrained_model/zephyr-7b-beta"
55
  tokenizer_path = "HuggingFaceH4/zephyr-7b-beta"
56
  tokenizer = AutoTokenizer.from_pretrained(
57
  tokenizer_path, add_prefix_space=False, use_fast=True, trust_remote_code=True, add_eos_token=True