elplaguister commited on
Commit
717cb6d
β€’
1 Parent(s): caaf812

Update koalpaca.py

Browse files
Files changed (1) hide show
  1. 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('cuda'),
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]