yuhaofeng-shiba commited on
Commit
b35cff2
·
1 Parent(s): 1561d65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ def init_args():
22
  args.batch_size = 1
23
  args.seq_length = 1024
24
  args.world_size = 1
25
- args.use_int8 = True
26
  args.top_p = 0
27
  args.repetition_penalty_range = 1024
28
  args.repetition_penalty_slope = 0
@@ -38,7 +38,7 @@ def init_model():
38
  global lm_generation
39
  torch.set_default_tensor_type(torch.HalfTensor)
40
  model = LLaMa(args)
41
- torch.set_default_tensor_type(torch.FloatTensor)
42
  model = load_model(model, args.load_model_path)
43
  model.eval()
44
 
 
22
  args.batch_size = 1
23
  args.seq_length = 1024
24
  args.world_size = 1
25
+ args.use_int8 = False
26
  args.top_p = 0
27
  args.repetition_penalty_range = 1024
28
  args.repetition_penalty_slope = 0
 
38
  global lm_generation
39
  torch.set_default_tensor_type(torch.HalfTensor)
40
  model = LLaMa(args)
41
+ # torch.set_default_tensor_type(torch.FloatTensor)
42
  model = load_model(model, args.load_model_path)
43
  model.eval()
44