isayahc commited on
Commit
336cf54
·
1 Parent(s): 959ed33

attempt to initialize model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,10 +26,10 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
26
 
27
 
28
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=350, chunk_overlap=10)
29
- model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
30
  embeddings = HuggingFaceHubEmbeddings()
31
 
32
- # model_id = "HuggingFaceH4/zephyr-7b-beta"
33
  # model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
34
  model = AutoModelForCausalLM.from_pretrained(
35
  model_id,
 
26
 
27
 
28
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=350, chunk_overlap=10)
29
+ # model_id = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
30
  embeddings = HuggingFaceHubEmbeddings()
31
 
32
+ model_id = "HuggingFaceH4/zephyr-7b-beta"
33
  # model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
34
  model = AutoModelForCausalLM.from_pretrained(
35
  model_id,