KaizeShi commited on
Commit
6301277
·
1 Parent(s): 8b567c4

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ else:
56
  device_map={"": device},
57
  low_cpu_mem_usage=True
58
  )
59
- print("Model: " + str(model))
60
  model = PeftModel.from_pretrained(
61
  model,
62
  LORA_WEIGHTS,
 
56
  device_map={"": device},
57
  low_cpu_mem_usage=True
58
  )
59
+ print("LLaMA Loaded")
60
  model = PeftModel.from_pretrained(
61
  model,
62
  LORA_WEIGHTS,