Hugging Face's logo

Spaces:
ignitariumcloud
/
Apollo_GenAI
Runtime error

App Files Community
VishalD1234 commited on
Commit
d064df9
·
verified ·
1 Parent(s): 33101ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  from PIL import Image
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
6
 
7
- MODEL_PATH = "THUDM/cogvlm2-llama3-caption"
8
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
9
  TORCH_TYPE = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_capability()[0] >= 8 else torch.float16
10
 
 
4
  from PIL import Image
5
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
6
 
7
+ MODEL_PATH = "THUDM/cogvlm2-video-llama3-chat"
8
  DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
9
  TORCH_TYPE = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_capability()[0] >= 8 else torch.float16
10