Spaces:
Runtime error
Runtime error
attempt to initialize model
Browse files
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 |
-
|
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,
|