MisterAI commited on
Commit
b7a52e3
·
verified ·
1 Parent(s): f3d1e8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -62,7 +62,9 @@ model_id = "mistralai/Mistral-Nemo-Instruct-2407"
62
 
63
  # Initialisation du modèle avec des paramètres de contexte plus grands
64
  tokenizer = AutoTokenizer.from_pretrained(model_id)
65
- model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", token=token)
 
 
66
 
67
  def parse_presentation_content(content):
68
  """Parse le contenu généré en sections pour les diapositives"""
 
62
 
63
  # Initialisation du modèle avec des paramètres de contexte plus grands
64
  tokenizer = AutoTokenizer.from_pretrained(model_id)
65
+ #model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", token=token)
66
+ model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto")
67
+
68
 
69
  def parse_presentation_content(content):
70
  """Parse le contenu généré en sections pour les diapositives"""