gregorkrzmanc commited on
Commit
a02ee35
·
1 Parent(s): 4c79858
Files changed (1) hide show
  1. src/model_wrapper_gradio.py +1 -1
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/models/LGATr/lgatr.py"
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