Spaces:
Sleeping
Sleeping
Commit
·
a02ee35
1
Parent(s):
4c79858
src/model_wrapper_gradio.py
CHANGED
|
@@ -41,7 +41,7 @@ def inference(loss_str, train_dataset_str, input_text, input_text_quarks):
|
|
| 41 |
args.spatial_part_only = True # LGATr
|
| 42 |
args.load_model_weights = model_path
|
| 43 |
args.aug_soft = True # LGATr_GP etc.
|
| 44 |
-
args.network_config = "src/
|
| 45 |
args.beta_type = "pt+bc"
|
| 46 |
args.embed_as_vectors = False
|
| 47 |
args.debug = False
|
|
|
|
| 41 |
args.spatial_part_only = True # LGATr
|
| 42 |
args.load_model_weights = model_path
|
| 43 |
args.aug_soft = True # LGATr_GP etc.
|
| 44 |
+
args.network_config = "src/1models/LGATr/lgatr.py"
|
| 45 |
args.beta_type = "pt+bc"
|
| 46 |
args.embed_as_vectors = False
|
| 47 |
args.debug = False
|