Spaces:

Futuretop
/

CaricatureGenerator-4.0

Runtime error

Futuretop commited on May 22

Commit

b715851

verified ·

1 Parent(s): 9f92f81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -87,7 +87,8 @@ tokenizer         = CLIPTokenizer.from_pretrained(
     model_repo, subfolder="tokenizer"
 )
 feature_extractor = CLIPFeatureExtractor.from_pretrained(
-    model_repo, subfolder="feature_extractor"
 )
 unet = UNet2DConditionModel.from_pretrained(
     model_repo, subfolder="unet", torch_dtype=torch.float16

     model_repo, subfolder="tokenizer"
 )
 feature_extractor = CLIPFeatureExtractor.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    subfolder="feature_extractor"
 )
 unet = UNet2DConditionModel.from_pretrained(
     model_repo, subfolder="unet", torch_dtype=torch.float16