Spaces:
Runtime error
Runtime error
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")
|