Skip to content

Commit

Permalink
Update docs
Browse files Browse the repository at this point in the history
  • Loading branch information
charlessjc committed Oct 8, 2024
1 parent 580eece commit a859cf0
Showing 3 changed files with 5 additions and 3 deletions.
4 changes: 2 additions & 2 deletions docs/api.rst
Original file line number Diff line number Diff line change
@@ -58,12 +58,12 @@ libcom.fopa_heat_map

libcom.shadow_generation
----------------------------------------
.. autoclass:: libcom.ShadowGenerationModel
.. automodule:: libcom.shadow_generation
:members:
:special-members: __call__

libcom.objectstitch
----------------------------------------
.. autoclass:: libcom.Mure_ObjectStitchModel
.. automodule:: libcom.objectstitch
:members:
:special-members: __call__
1 change: 1 addition & 0 deletions libcom/shadow_generation/shadow_generation.py
Original file line number Diff line number Diff line change
@@ -101,6 +101,7 @@ def __init__(self, device=0, model_type='ShadowGeneration', **kwargs):
download_pretrained_model(cldm_weight_path)
download_pretrained_model(cls_weight_path)
download_pretrained_model(reg_weight_path)
download_pretrained_model(cls_label_path)
self.device = check_gpu_device(device)
self.build_pretrained_model(ppp_weight_path, cldm_weight_path, cls_weight_path, reg_weight_path, cls_label_path)
self.build_data_transformer()
3 changes: 2 additions & 1 deletion requirements/runtime.txt
Original file line number Diff line number Diff line change
@@ -21,4 +21,5 @@ streamlit==1.20.0
timm==0.9.8
tqdm==4.65.0
transformers==4.27.4
open_clip_torch==2.7.0
open_clip_torch==2.7.0
diffusers==0.29.2

0 comments on commit a859cf0

Please sign in to comment.