Spaces:

JingyeChen22
/

TextDiffuser-2

Running on A10G

JingyeChen commited on Dec 10, 2023

Commit

1781379

•

1 Parent(s): 0a8222c

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ m1_model = AutoModelForCausalLM.from_pretrained(
 #### import diffusion models
 text_encoder = CLIPTextModel.from_pretrained(
-    'JingyeChen22/textdiffuser2-full-ft', subfolder="text_encoder", ignore_mismatched_sizes=True
 ).cuda().half()
 tokenizer = CLIPTokenizer.from_pretrained(
     'runwayml/stable-diffusion-v1-5', subfolder="tokenizer"

 #### import diffusion models
 text_encoder = CLIPTextModel.from_pretrained(
+    'JingyeChen22/textdiffuser2-full-ft', subfolder="text_encoder"
 ).cuda().half()
 tokenizer = CLIPTokenizer.from_pretrained(
     'runwayml/stable-diffusion-v1-5', subfolder="tokenizer"