Spaces:

Lwasinam
/

image-caption

Sleeping

Lwasinam commited on Mar 27, 2024

Commit

f9a92b5

verified ·

1 Parent(s): 705b7e0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -170,7 +170,7 @@ def main():
         tokenizer = get_or_build_tokenizer(config)
         model = get_model(config, len(tokenizer))
         model = accelerator.prepare(model)
-        accelerator.load_state('model.safetensors')
         # model = get_model(config, len(tokenizer))
         # model.to(device)

         tokenizer = get_or_build_tokenizer(config)
         model = get_model(config, len(tokenizer))
         model = accelerator.prepare(model)
+        accelerator.load_state('')
         # model = get_model(config, len(tokenizer))
         # model.to(device)