Spaces:
Running
Running
Akash Garg
commited on
Commit
·
487ce21
1
Parent(s):
4586de6
change to use engine
Browse files
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 =
|
| 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 = Engine(
|
| 125 |
config_path,
|
| 126 |
gpt_ckpt_path,
|
| 127 |
shape_ckpt_path,
|