rafavidal1709 commited on
Commit
364da20
·
verified ·
1 Parent(s): 22c7a6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -8,8 +8,7 @@ MODEL_NAME = "deepseek-ai/DeepSeek-R1"
8
  # Configuração segura para CPU
9
  tokenizer = AutoTokenizer.from_pretrained(
10
  MODEL_NAME,
11
- trust_remote_code=True,
12
- revision="6528ae3" # Fixar versão específica
13
  )
14
 
15
  model = AutoModelForCausalLM.from_pretrained(
@@ -19,7 +18,7 @@ model = AutoModelForCausalLM.from_pretrained(
19
  device_map="cpu",
20
  low_cpu_mem_usage=True,
21
  load_in_8bit=False,
22
- offload_folder="offload" # Pasta para descarregar pesos grandes
23
  )
24
 
25
  def summarize_text(text):
 
8
  # Configuração segura para CPU
9
  tokenizer = AutoTokenizer.from_pretrained(
10
  MODEL_NAME,
11
+ trust_remote_code=True # Removendo a revisão específica
 
12
  )
13
 
14
  model = AutoModelForCausalLM.from_pretrained(
 
18
  device_map="cpu",
19
  low_cpu_mem_usage=True,
20
  load_in_8bit=False,
21
+ offload_folder="offload"
22
  )
23
 
24
  def summarize_text(text):