Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ job_count = 1
|
|
| 34 |
|
| 35 |
|
| 36 |
def device():
|
| 37 |
-
return
|
| 38 |
|
| 39 |
|
| 40 |
def generate_batch(input_ids, min_length, max_length, num_beams, temperature, repetition_penalty, length_penalty, sampling_mode, top_k, top_p):
|
|
|
|
| 34 |
|
| 35 |
|
| 36 |
def device():
|
| 37 |
+
return torch.device("cpu")
|
| 38 |
|
| 39 |
|
| 40 |
def generate_batch(input_ids, min_length, max_length, num_beams, temperature, repetition_penalty, length_penalty, sampling_mode, top_k, top_p):
|