SunDay-s commited on
Commit
516b2a7
·
verified ·
1 Parent(s): 3ff7679

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -2,10 +2,11 @@ import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
- # Загрузка модели и токенизатора с доверием к пользовательскому коду
6
  model_name = "deepseek-ai/DeepSeek-V3"
7
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
8
- model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
9
 
10
  # Системное сообщение
11
  system_message = (
 
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
+ # Загрузка модели и токенизатора с фиксированной ревизией
6
  model_name = "deepseek-ai/DeepSeek-V3"
7
+ revision = "main" # Замените на конкретную ревизию
8
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True, revision=revision)
9
+ model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, revision=revision)
10
 
11
  # Системное сообщение
12
  system_message = (