NotYuSheng commited on
Commit
33a7ee6
1 Parent(s): 0aa318f

Update multimodal_ai.py

Browse files
Files changed (1) hide show
  1. multimodal_ai.py +1 -1
multimodal_ai.py CHANGED
@@ -17,7 +17,7 @@ class MultimodalAI:
17
  login(token=self.HUGGINGFACE_TOKEN)
18
 
19
  # Model selection
20
- self.model_name = "meta-llama/Meta-Llama-3-8B"
21
 
22
  # Check if a CUDA-enabled GPU is available.
23
  # If available, move the model to the GPU (cuda:0) for faster computation.
 
17
  login(token=self.HUGGINGFACE_TOKEN)
18
 
19
  # Model selection
20
+ self.model_name = "meta-llama/Llama-2-7b-hf"
21
 
22
  # Check if a CUDA-enabled GPU is available.
23
  # If available, move the model to the GPU (cuda:0) for faster computation.