Spaces:

yeliudev
/

R2-Tuning

Running

yeliudev commited on Mar 29

Commit

4503b36

verified ·

1 Parent(s): f948fcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def init_model(config, checkpoint):
     cfg.model.init = True
     if checkpoint.startswith('http'):
-        checkpoint = nncore.download(checkpoint, out_dir='checkpoints')
     model = build_model(cfg.model, dist=False).eval()
     model = load_checkpoint(model, checkpoint, warning=False)

     cfg.model.init = True
     if checkpoint.startswith('http'):
+        checkpoint = nncore.download(checkpoint, out_dir='checkpoints', verbose=False)
     model = build_model(cfg.model, dist=False).eval()
     model = load_checkpoint(model, checkpoint, warning=False)