ahmedfaiyaz
/

OkkhorDiffusion-CMATERdb

OkkhorDiffusionPipeline

Model card Files Files and versions Community

ahmedfaiyaz commited on Feb 28

Commit

0975e51

•

1 Parent(s): fa40e0b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ device="cuda"
 pipeline = DiffusionPipeline.from_pretrained(
               "ahmedfaiyaz/OkkhorDiffusion-CMATERdb",
               custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",
-              embedding=torch.float16
             )
 pipeline.to(device)
 pipeline.embedding=torch.tensor([0],device=device) # 'প্র': 0

 pipeline = DiffusionPipeline.from_pretrained(
               "ahmedfaiyaz/OkkhorDiffusion-CMATERdb",
               custom_pipeline="ahmedfaiyaz/OkkhorDiffusion",
+              embedding=torch.int16
             )
 pipeline.to(device)
 pipeline.embedding=torch.tensor([0],device=device) # 'প্র': 0