Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SHI-Labs/Forget-Me-Not
/ types & classes
Types & classes
4 in github.com/SHI-Labs/Forget-Me-Not
⨍
Functions
31
◇
Types & classes
4
↓ 1 callers
Class
AttnController
src/train_attn.py:279
↓ 1 callers
Class
ForgetMeNotDataset
A dataset to prepare the instance and class images with the prompts for fine-tuning the model. It pre-processes the images and the tokenizes
src/train_attn.py:59
↓ 1 callers
Class
MyCrossAttnProcessor
src/train_attn.py:298
↓ 1 callers
Class
PivotalTuningDatasetCapation
A dataset to prepare the instance and class images with the prompts for fine-tuning the model. It pre-processes the images and the tokenizes
src/dataset.py:33