Spaces:
Sleeping
Sleeping
rynmurdock
commited on
Commit
•
29b27a4
1
Parent(s):
a8fb4ec
Update app.py
Browse files
app.py
CHANGED
@@ -92,13 +92,10 @@ device_map='cuda')
|
|
92 |
# vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=dtype)
|
93 |
# vae = compile_unet(vae, config=config)
|
94 |
|
95 |
-
#finetune_path = '''/home/ryn_mote/Misc/finetune-sd1.5/dreambooth-model best'''''
|
96 |
-
#unet = UNet2DConditionModel.from_pretrained(finetune_path+'/unet/').to(dtype)
|
97 |
-
#text_encoder = CLIPTextModel.from_pretrained(finetune_path+'/text_encoder/').to(dtype)
|
98 |
|
99 |
|
100 |
-
unet = UNet2DConditionModel.from_pretrained('
|
101 |
-
text_encoder = CLIPTextModel.from_pretrained('
|
102 |
device_map='cpu').to(dtype)
|
103 |
|
104 |
adapter = MotionAdapter.from_pretrained("wangfuyun/AnimateLCM")
|
|
|
92 |
# vae = ConsistencyDecoderVAE.from_pretrained("openai/consistency-decoder", torch_dtype=dtype)
|
93 |
# vae = compile_unet(vae, config=config)
|
94 |
|
|
|
|
|
|
|
95 |
|
96 |
|
97 |
+
unet = UNet2DConditionModel.from_pretrained('emilianJR/epiCRealism', subfolder='unet',).to(dtype).to('cpu')
|
98 |
+
text_encoder = CLIPTextModel.from_pretrained('emilianJR/epiCRealism', subfolder='text_encoder',
|
99 |
device_map='cpu').to(dtype)
|
100 |
|
101 |
adapter = MotionAdapter.from_pretrained("wangfuyun/AnimateLCM")
|