Spaces:
Sleeping
Sleeping
home pc cache dir added
Browse files
app.py
CHANGED
@@ -19,12 +19,15 @@ login(os.environ["HF_TOKEN"])
|
|
19 |
dt = datetime.datetime.now()
|
20 |
print(dt)
|
21 |
print("loading models")
|
22 |
-
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
|
23 |
-
original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
|
24 |
-
untethered_model = GPT2LMHeadModel.from_pretrained('zmbfeng/untethered_20240225_epochs_500')
|
25 |
# tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
26 |
# original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
27 |
# untethered_model = GPT2LMHeadModel.from_pretrained('zmbfeng/untethered_20240225_epochs_500',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
|
|
|
|
|
|
28 |
|
29 |
default_temperature=0.01
|
30 |
default_seed=43
|
|
|
19 |
dt = datetime.datetime.now()
|
20 |
print(dt)
|
21 |
print("loading models")
|
22 |
+
# tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
|
23 |
+
# original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
|
24 |
+
# untethered_model = GPT2LMHeadModel.from_pretrained('zmbfeng/untethered_20240225_epochs_500')
|
25 |
# tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
26 |
# original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
27 |
# untethered_model = GPT2LMHeadModel.from_pretrained('zmbfeng/untethered_20240225_epochs_500',cache_dir="G:\My Drive\Avatar\language_models_windows")
|
28 |
+
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium',cache_dir="C:\\Users\\zmbfeng\\Google Drive\\language_models_windows")
|
29 |
+
original_model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium',cache_dir="C:\\Users\\zmbfeng\\Google Drive\\Avatar\\language_models_windows")
|
30 |
+
untethered_model = GPT2LMHeadModel.from_pretrained('zmbfeng/untethered_20240225_epochs_500',cache_dir="C:\\Users\\zmbfeng\\Google Drive\\Avatar\\language_models_windows")
|
31 |
|
32 |
default_temperature=0.01
|
33 |
default_seed=43
|