p1atdev
/

wd-swinv2-tagger-v3-hf

Image Classification

Model card Files Files and versions Community

p1atdev commited on Mar 8, 2024

Commit

1f728f2

·

verified ·

1 Parent(s): d6fd8d7

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -60,7 +60,6 @@ MODEL_NAME = "p1atdev/wd-swinv2-tagger-v3-hf"
 model = AutoModelForImageClassification.from_pretrained(
     MODEL_NAME,
-    torch_dtype=torch.bfloat16,
 )
 model.eval()
 processor = AutoImageProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)

 model = AutoModelForImageClassification.from_pretrained(
     MODEL_NAME,
 )
 model.eval()
 processor = AutoImageProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)