DeepSeek/DreamCraft3D
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DreamCraft3D synced 2024-12-04 18:15:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove local_files_only

Browse Source
This commit is contained in:
MrTornado24 2023-12-21 13:04:45 +08:00
parent 624aa2b8ce
commit 0ae1a4dc2b
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
threestudio/models/prompt_processors/deepfloyd_prompt_processor.py
View File

@ -58,7 +58,6 @@ class DeepFloydPromptProcessor(PromptProcessor):
tokenizer = T5Tokenizer.from_pretrained(
pretrained_model_name_or_path,
subfolder="tokenizer",
local_files_only=True
)
text_encoder = T5EncoderModel.from_pretrained(
pretrained_model_name_or_path,
@ -67,7 +66,6 @@ class DeepFloydPromptProcessor(PromptProcessor):
load_in_8bit=True,
variant="8bit",
device_map="auto",
local_files_only=True
)
with torch.no_grad():
text_inputs = tokenizer(

2
threestudio/models/prompt_processors/stable_diffusion_prompt_processor.py
View File

@ -75,13 +75,11 @@ class StableDiffusionPromptProcessor(PromptProcessor):
tokenizer = AutoTokenizer.from_pretrained(
pretrained_model_name_or_path,
subfolder="tokenizer",
local_files_only=True,
)
text_encoder = CLIPTextModel.from_pretrained(
pretrained_model_name_or_path,
subfolder="text_encoder",
device_map="auto",
local_files_only=True,
)
with torch.no_grad():