Kuberwastaken commited on
Commit
9da5a57
·
1 Parent(s): 472c34c

Added trust_remote_code=True

Browse files
Files changed (1) hide show
  1. model/analyzer.py +4 -2
model/analyzer.py CHANGED
@@ -62,7 +62,8 @@ class ContentAnalyzer:
62
 
63
  self.tokenizer = AutoTokenizer.from_pretrained(
64
  "LGAI-EXAONE/EXAONE-Deep-2.4B",
65
- use_fast=True
 
66
  )
67
 
68
  if progress:
@@ -71,7 +72,8 @@ class ContentAnalyzer:
71
  self.model = AutoModelForSeq2SeqLM.from_pretrained(
72
  "LGAI-EXAONE/EXAONE-Deep-2.4B",
73
  torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
74
- device_map="auto"
 
75
  )
76
 
77
  if self.device == "cuda":
 
62
 
63
  self.tokenizer = AutoTokenizer.from_pretrained(
64
  "LGAI-EXAONE/EXAONE-Deep-2.4B",
65
+ use_fast=True,
66
+ trust_remote_code=True
67
  )
68
 
69
  if progress:
 
72
  self.model = AutoModelForSeq2SeqLM.from_pretrained(
73
  "LGAI-EXAONE/EXAONE-Deep-2.4B",
74
  torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
75
+ device_map="auto",
76
+ trust_remote_code=True
77
  )
78
 
79
  if self.device == "cuda":