Akash Garg commited on
Commit
f641176
·
1 Parent(s): 0c04a17

engine fast

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -121,7 +121,7 @@ if __name__=="__main__":
121
  config_path = args.config_path
122
  gpt_ckpt_path = "./model_weights/shape_gpt.safetensors"
123
  shape_ckpt_path = "./model_weights/shape_tokenizer.safetensors"
124
- engine_fast = Engine(
125
  config_path,
126
  gpt_ckpt_path,
127
  shape_ckpt_path,
 
121
  config_path = args.config_path
122
  gpt_ckpt_path = "./model_weights/shape_gpt.safetensors"
123
  shape_ckpt_path = "./model_weights/shape_tokenizer.safetensors"
124
+ engine_fast = EngineFast(
125
  config_path,
126
  gpt_ckpt_path,
127
  shape_ckpt_path,