MOSS550V commited on
Commit
75e2698
·
1 Parent(s): 96fe4c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ def main():
130
  config.pre_seq_len = 128
131
  config.prefix_projection = False
132
 
133
- ptuning_checkpoint = "MOSS550V/divination"
134
 
135
  if ptuning_checkpoint is not None:
136
  print(f"Loading prefix_encoder weight from {ptuning_checkpoint}")
 
130
  config.pre_seq_len = 128
131
  config.prefix_projection = False
132
 
133
+ ptuning_checkpoint = "checkpoint-500"
134
 
135
  if ptuning_checkpoint is not None:
136
  print(f"Loading prefix_encoder weight from {ptuning_checkpoint}")