Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,10 @@ gpt2_lm = keras_nlp.models.GPT2CausalLM.from_preset(
|
|
| 41 |
)
|
| 42 |
|
| 43 |
#drive.mount('/content/drive', force_remount=True)
|
|
|
|
| 44 |
checkpoint_path = "./aloqas_model_checkpoints/cp.ckpt"
|
|
|
|
|
|
|
| 45 |
|
| 46 |
gpt2_lm.load_weights(checkpoint_path)
|
| 47 |
|
|
|
|
| 41 |
)
|
| 42 |
|
| 43 |
#drive.mount('/content/drive', force_remount=True)
|
| 44 |
+
# GPT2 Base 1 epochs
|
| 45 |
checkpoint_path = "./aloqas_model_checkpoints/cp.ckpt"
|
| 46 |
+
# GPT2 Base 3 epochs
|
| 47 |
+
checkpoint_path = "./aloqas_model_checkpoints_gpt2base_3_epochs/cp.ckpt"
|
| 48 |
|
| 49 |
gpt2_lm.load_weights(checkpoint_path)
|
| 50 |
|