Hugging Face's logo

Spaces:
suayptalha
/
Chat-with-Bitnet-b1.58-2B-4T
Running on Zero

App Files Community
suayptalha commited on
Commit
7b31c01
·
verified ·
1 Parent(s): baba7b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -19,7 +19,6 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_id,
21
  torch_dtype=torch.bfloat16,
22
- device_map="auto"
23
  )
24
 
25
  @spaces.GPU
 
19
  model = AutoModelForCausalLM.from_pretrained(
20
  model_id,
21
  torch_dtype=torch.bfloat16,
 
22
  )
23
 
24
  @spaces.GPU