rphrp1985 commited on
Commit
8faa079
·
verified ·
1 Parent(s): 504635f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -30,14 +30,15 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
30
  model_id = "CohereForAI/c4ai-command-r-plus-4bit"
31
 
32
 
33
- tokenizer = AutoTokenizer.from_pretrained(model_id, token= token)
34
 
35
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,
36
  # torch_dtype=torch.bfloat16,
37
  attn_implementation="flash_attention_2",
38
  # low_cpu_mem_usage=True,
39
  # llm_int8_enable_fp32_cpu_offload=True,
40
- device_map="cuda"
 
41
  )
42
 
43
 
 
30
  model_id = "CohereForAI/c4ai-command-r-plus-4bit"
31
 
32
 
33
+ tokenizer = AutoTokenizer.from_pretrained(model_id, token= token, quantization=False)
34
 
35
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,
36
  # torch_dtype=torch.bfloat16,
37
  attn_implementation="flash_attention_2",
38
  # low_cpu_mem_usage=True,
39
  # llm_int8_enable_fp32_cpu_offload=True,
40
+ device_map="cuda",
41
+ quantization=False
42
  )
43
 
44