Yaron Koresh commited on
Commit
889b1ec
·
verified ·
1 Parent(s): 8d20a09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -21,7 +21,7 @@ dtype = torch.float16
21
  step = 2
22
  repo = "ByteDance/AnimateDiff-Lightning"
23
  ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
24
- model_id = "black-forest-labs/FLUX.1-schnell"
25
 
26
  adapter = MotionAdapter().to(device, dtype)
27
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
@@ -45,14 +45,6 @@ pipe.scheduler = scheduler
45
  pipe.enable_vae_slicing()
46
  pipe.enable_vae_tiling()
47
 
48
- """
49
- flx = FluxPipeline.from_pretrained(
50
- base,
51
- torch_dtype=dtype,
52
- token=os.getenv("hf_token")
53
- ).to(device)
54
- """
55
-
56
  def translate(text,lang):
57
 
58
  if text == None or lang == None:
 
21
  step = 2
22
  repo = "ByteDance/AnimateDiff-Lightning"
23
  ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
24
+ model_id = "emilianJR/epiCRealism"
25
 
26
  adapter = MotionAdapter().to(device, dtype)
27
  adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
 
45
  pipe.enable_vae_slicing()
46
  pipe.enable_vae_tiling()
47
 
 
 
 
 
 
 
 
 
48
  def translate(text,lang):
49
 
50
  if text == None or lang == None: