katsukiai commited on
Commit
4bce260
·
verified ·
1 Parent(s): 630bd17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ logger = logging.getLogger(__name__)
14
  # Load DeepSeek-V3 model and tokenizer for CPU
15
  MODEL_NAME = "deepseek-ai/deepseek-v3"
16
  logger.info(f"Loading model: {MODEL_NAME} (CPU mode)")
17
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
18
- model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float32, device_map="cpu")
19
 
20
  # Function to process text with DeepSeek-V3
21
  def process_text_with_model(text):
 
14
  # Load DeepSeek-V3 model and tokenizer for CPU
15
  MODEL_NAME = "deepseek-ai/deepseek-v3"
16
  logger.info(f"Loading model: {MODEL_NAME} (CPU mode)")
17
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME,trust_remote_code=True)
18
+ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, trust_remote_code=True,torch_dtype=torch.float32, device_map="cpu")
19
 
20
  # Function to process text with DeepSeek-V3
21
  def process_text_with_model(text):