NotYuSheng commited on
Commit
d02e038
1 Parent(s): ff0c29a

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/Llama-2-7b"
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 = "Lin-Chen/sharegpt4video-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.