zhaozitian commited on
Commit
02dc13e
·
1 Parent(s): 806585b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ if device == "cuda":
30
  load_in_8bit=False,
31
  torch_dtype=torch.float16,
32
  device_map="auto",
33
- from_pretrained,
34
  )
35
  model = PeftModel.from_pretrained(
36
  model, LORA_WEIGHTS, torch_dtype=torch.float16, force_download=True
@@ -40,7 +40,7 @@ elif device == "mps":
40
  BASE_MODEL,
41
  device_map={"": device},
42
  torch_dtype=torch.float16,
43
- from_pretrained,
44
  )
45
  model = PeftModel.from_pretrained(
46
  model,
 
30
  load_in_8bit=False,
31
  torch_dtype=torch.float16,
32
  device_map="auto",
33
+ use_auth_token = True,
34
  )
35
  model = PeftModel.from_pretrained(
36
  model, LORA_WEIGHTS, torch_dtype=torch.float16, force_download=True
 
40
  BASE_MODEL,
41
  device_map={"": device},
42
  torch_dtype=torch.float16,
43
+ use_auth_token = True,
44
  )
45
  model = PeftModel.from_pretrained(
46
  model,