Zoe911 commited on
Commit
eff8c08
ยท
verified ยท
1 Parent(s): 2e2aed3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
14
  torch_dtype=torch.bfloat16,
15
  device_map="auto" # ้œ€่ฆaccelerateๆ”ฏๆŒ
16
  )
17
- # ๅฆ‚ๆžœๆœ‰ๅพฎ่ฐƒๆจกๅž‹๏ผŒๅ–ๆถˆๆณจ้‡Šไปฅไธ‹่กŒ
18
  # model = PeftModel.from_pretrained(base_model, lora_model_name)
19
  model = base_model # ๆš‚ๆ—ถไฝฟ็”จๅŸบ็ก€ๆจกๅž‹
20
 
 
14
  torch_dtype=torch.bfloat16,
15
  device_map="auto" # ้œ€่ฆaccelerateๆ”ฏๆŒ
16
  )
17
+
18
  # model = PeftModel.from_pretrained(base_model, lora_model_name)
19
  model = base_model # ๆš‚ๆ—ถไฝฟ็”จๅŸบ็ก€ๆจกๅž‹
20