Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,6 +28,7 @@ from fudge.util import save_checkpoint, ProgressMeter, AverageMeter, num_params
|
|
| 28 |
from fudge.constants import *
|
| 29 |
|
| 30 |
|
|
|
|
| 31 |
# imp.reload(model)
|
| 32 |
pretrained_model = "checkpoint-150/"
|
| 33 |
generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
|
|
|
|
| 28 |
from fudge.constants import *
|
| 29 |
|
| 30 |
|
| 31 |
+
device = 'cuda'
|
| 32 |
# imp.reload(model)
|
| 33 |
pretrained_model = "checkpoint-150/"
|
| 34 |
generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
|