Spaces:
Running
on
L40S
Running
on
L40S
Akash Garg
commited on
Commit
·
f641176
1
Parent(s):
0c04a17
engine fast
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 = EngineFast(
|
125 |
config_path,
|
126 |
gpt_ckpt_path,
|
127 |
shape_ckpt_path,
|