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]
|