Spaces:
Sleeping
Sleeping
anilbhatt1
commited on
Commit
•
6b9015d
1
Parent(s):
0f48d1a
Update app.py
Browse files
app.py
CHANGED
@@ -5,11 +5,11 @@ def generate_text(context, num_samples, context_length, model_name):
|
|
5 |
from pathlib import Path
|
6 |
|
7 |
if model_name == "pythia_160m_deduped_custom" or model_name == "pythia_160m_deduped_huggingface":
|
8 |
-
ckpt_dir = Path('
|
9 |
elif model_name == "pythia_70m_deduped":
|
10 |
-
ckpt_dir = Path('
|
11 |
elif model_name == "pythia_410m_deduped":
|
12 |
-
ckpt_dir = Path('
|
13 |
|
14 |
context = str(context)
|
15 |
num_samples = int(num_samples)
|
|
|
5 |
from pathlib import Path
|
6 |
|
7 |
if model_name == "pythia_160m_deduped_custom" or model_name == "pythia_160m_deduped_huggingface":
|
8 |
+
ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-160m-deduped')
|
9 |
elif model_name == "pythia_70m_deduped":
|
10 |
+
ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-70m-deduped')
|
11 |
elif model_name == "pythia_410m_deduped":
|
12 |
+
ckpt_dir = Path('/home/user/app/checkpoints/EleutherAI/pythia-410m-deduped')
|
13 |
|
14 |
context = str(context)
|
15 |
num_samples = int(num_samples)
|