bombaygamercc commited on
Commit
fd0d296
1 Parent(s): 5334440

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -35,9 +35,9 @@ ALLOWED_VIDEO_EXTENSIONS = {'mp4', 'avi', 'mov', 'mkv'}
35
  ALLOWED_AUDIO_EXTENSIONS = {'wav', 'mp3', 'm4a', 'flac'}
36
 
37
  # Initialize Mistral 7B model and tokenizer
38
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", trust_remote_code=True)
39
  model = AutoModelForCausalLM.from_pretrained(
40
- "mistralai/Mistral-7B-v0.1",
41
  torch_dtype="auto",
42
  device_map="auto",
43
  trust_remote_code=True
 
35
  ALLOWED_AUDIO_EXTENSIONS = {'wav', 'mp3', 'm4a', 'flac'}
36
 
37
  # Initialize Mistral 7B model and tokenizer
38
+ tokenizer = AutoTokenizer.from_pretrained("bombaygamercc/learnyfi-mistral", trust_remote_code=True)
39
  model = AutoModelForCausalLM.from_pretrained(
40
+ "bombaygamercc/learnyfi-mistral",
41
  torch_dtype="auto",
42
  device_map="auto",
43
  trust_remote_code=True