Diffusers
Safetensors
PixCellControlNetPipeline
AlexGraikos commited on
Commit
f5e2ae6
·
verified ·
1 Parent(s): fa7e6ae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ timm_kwargs = {
47
  'dynamic_img_size': True
48
  }
49
  uni_model = timm.create_model("hf-hub:MahmoodLab/UNI2-h", pretrained=True, **timm_kwargs)
50
- transform = create_transform(**resolve_data_config(uni_model.pretrained_cfg, model=uni_model))
51
  uni_model.eval()
52
  uni_model.to(device);
53
  ```
 
47
  'dynamic_img_size': True
48
  }
49
  uni_model = timm.create_model("hf-hub:MahmoodLab/UNI2-h", pretrained=True, **timm_kwargs)
50
+ uni_transforms = create_transform(**resolve_data_config(uni_model.pretrained_cfg, model=uni_model))
51
  uni_model.eval()
52
  uni_model.to(device);
53
  ```