Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ db = Chroma(persist_directory=CHROMA_PATH, embedding_function=embeddings)
|
|
91 |
|
92 |
|
93 |
|
94 |
-
MODEL_NAME = "
|
95 |
|
96 |
model = AutoModelForCausalLM.from_pretrained(
|
97 |
MODEL_NAME, trust_remote_code=True, device_map="auto",offload_folder="offload"
|
|
|
91 |
|
92 |
|
93 |
|
94 |
+
MODEL_NAME = "gpt2"
|
95 |
|
96 |
model = AutoModelForCausalLM.from_pretrained(
|
97 |
MODEL_NAME, trust_remote_code=True, device_map="auto",offload_folder="offload"
|