Xkev commited on
Commit
929ec82
·
verified ·
1 Parent(s): d6e043f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -9,6 +9,8 @@ from gradio import FileData
9
  import time
10
  import spaces
11
  import re
 
 
12
  ckpt = "Xkev/Llama-3.2V-11B-cot"
13
  model = MllamaForConditionalGeneration.from_pretrained(ckpt,
14
  torch_dtype=torch.bfloat16).to("cuda")
 
9
  import time
10
  import spaces
11
  import re
12
+ import copy
13
+
14
  ckpt = "Xkev/Llama-3.2V-11B-cot"
15
  model = MllamaForConditionalGeneration.from_pretrained(ckpt,
16
  torch_dtype=torch.bfloat16).to("cuda")