rynmurdock commited on
Commit
29b27a4
1 Parent(s): a8fb4ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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('rynmurdock/Sea_Claws', subfolder='unet',).to(dtype).to('cpu')
101
- text_encoder = CLIPTextModel.from_pretrained('rynmurdock/Sea_Claws', subfolder='text_encoder',
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")