Spaces:
Paused
Paused
elplaguister commited on
Commit ·
717cb6d
1
Parent(s): caaf812
Update koalpaca.py
Browse files- koalpaca.py +1 -1
koalpaca.py
CHANGED
|
@@ -28,7 +28,7 @@ class KoAlpaca(Model):
|
|
| 28 |
inputs,
|
| 29 |
return_tensors='pt',
|
| 30 |
return_token_type_ids=False
|
| 31 |
-
).to('
|
| 32 |
generation_config=self.gen_config
|
| 33 |
)
|
| 34 |
outputs = self.tokenizer.decode(output_ids[0]).split("### 답변: ")[-1]
|
|
|
|
| 28 |
inputs,
|
| 29 |
return_tensors='pt',
|
| 30 |
return_token_type_ids=False
|
| 31 |
+
).to('cpu'),
|
| 32 |
generation_config=self.gen_config
|
| 33 |
)
|
| 34 |
outputs = self.tokenizer.decode(output_ids[0]).split("### 답변: ")[-1]
|