zmbfeng commited on
Commit
c64196b
·
1 Parent(s): d72eb57

home pc cache dir added

Browse files
Files changed (1) hide show
  1. app.py +6 -3
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