You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I didn't try, but i will... edit demo_textdiffuser2_inpainting_full.py
add
fromfastchat.modelimportget_conversation_templatefromtransformersimportCLIPTextModel, CLIPTokenizer, AutoTokenizer, AutoModelForCausalLMm1_model_path="JingyeChen22/textdiffuser2_layout_planner"cache_dir="model_cache"local_files_only=True# set to True if the models are saved in cache_dirm1_tokenizer=AutoTokenizer.from_pretrained(
m1_model_path,
use_fast=False,
cache_dir=cache_dir,
local_files_only=local_files_only,
)
m1_model=AutoModelForCausalLM.from_pretrained(
m1_model_path,
torch_dtype=torch.float16,
low_cpu_mem_usage=True,
cache_dir=cache_dir,
local_files_only=local_files_only,
).cuda()
Describe the bug
Model I am using : Textdiffuser2
The problem arises when using:
A clear and concise description of what the bug is.
demo_textdiffuser2_inpainting_full.py
doesn't work. huggingface link not exists.
and
def get_conversation_template()
m1_tokenizer, m1_model no exists
The text was updated successfully, but these errors were encountered: