debisoft commited on
Commit
4a9d3d0
·
1 Parent(s): 958e70f
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,8 +9,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
9
  from datasets import load_dataset
10
 
11
  huggingface_hub.login(os.getenv('HF_TOKEN'))
 
12
  #peft_model_id = "debisoft/Qwen2.5-VL-7B-Instruct-thinking-function_calling-quant-V0"
13
- peft_model_id = "debisoft/Qwen2.5-VL-3B-Instruct-thinking-function_calling-V0"
14
 
15
  bnb_config = BitsAndBytesConfig(
16
  load_in_4bit=True,
 
9
  from datasets import load_dataset
10
 
11
  huggingface_hub.login(os.getenv('HF_TOKEN'))
12
+ peft_model_id = "debisoft/DeepSeek-R1-Distill-Qwen-7B-thinking-function_calling-quant-V0"
13
  #peft_model_id = "debisoft/Qwen2.5-VL-7B-Instruct-thinking-function_calling-quant-V0"
14
+ #peft_model_id = "debisoft/Qwen2.5-VL-3B-Instruct-thinking-function_calling-V0"
15
 
16
  bnb_config = BitsAndBytesConfig(
17
  load_in_4bit=True,