Hugging Face's logo

Spaces:
openflamingo
/
OpenFlamingo
Runtime error

App Files Community
3
anas-awadalla commited on
Commit
b2e96cb
·
1 Parent(s): 64dbe9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ model, image_processor, tokenizer = create_model_and_transforms(
61
 
62
  checkpoint_path = hf_hub_download("openflamingo/OpenFlamingo-9B-vitl-mpt7b", "checkpoint.pt")
63
  model.load_state_dict(torch.load(checkpoint_path), strict=False)
64
- model = model.to("cuda")
65
  model.eval()
66
 
67
  def generate(
 
61
 
62
  checkpoint_path = hf_hub_download("openflamingo/OpenFlamingo-9B-vitl-mpt7b", "checkpoint.pt")
63
  model.load_state_dict(torch.load(checkpoint_path), strict=False)
64
+ model = model.to(0, dtype=torch.bfloat16)
65
  model.eval()
66
 
67
  def generate(