Dependency and import fixes

henk717 · Jan 23, 2024 · 7c48b0f · 7c48b0f
1 parent 974b967
commit 7c48b0f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 4 deletions.
diff --git a/environments/huggingface.yml b/environments/huggingface.yml
@@ -49,7 +49,7 @@ dependencies:
     - https://github.com/jllllll/bitsandbytes-windows-webui/releases/download/wheels/bitsandbytes-0.40.0.post4-py3-none-win_amd64.whl; sys_platform == 'win32'
     - ftfy
     - pydub
-    - diffusers
+    - diffusers==0.24
     - git+https://github.com/0cc4m/hf_bleeding_edge/
     - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux'
     - https://github.com/0cc4m/GPTQ-for-LLaMa/releases/download/0.0.6/gptq_koboldai-0.0.6-cp38-cp38-win_amd64.whl; sys_platform == 'win32'

diff --git a/environments/ipex.yml b/environments/ipex.yml
@@ -52,7 +52,7 @@ dependencies:
     - ijson
     - ftfy
     - pydub
-    - diffusers
+    - diffusers==0.24
     - git+https://github.com/0cc4m/hf_bleeding_edge/
     - https://huggingface.github.io/autogptq-index/whl/cu118/auto-gptq/auto_gptq-0.5.1%2Bcu118-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl; sys_platform == 'linux'
     - https://huggingface.github.io/autogptq-index/whl/cu118/auto-gptq/auto_gptq-0.5.1%2Bcu118-cp310-cp310-win_amd64.whl; sys_platform == 'win32'

diff --git a/environments/rocm.yml b/environments/rocm.yml
@@ -42,7 +42,7 @@ dependencies:
     - ijson
     - ftfy
     - pydub
-    - diffusers
+    - diffusers==0.24
     - git+https://github.com/0cc4m/hf_bleeding_edge/
     - einops
     - peft==0.7.1

diff --git a/modeling/inference_models/exllamav2/class.py b/modeling/inference_models/exllamav2/class.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 try:
+    import time, json
     import torch
     import numpy as np
     from typing import List, Optional, Union

diff --git a/requirements.txt b/requirements.txt
@@ -26,7 +26,7 @@ loguru
 termcolor
 git+https://github.com/VE-FORBRYDERNE/mkultra
 Pillow
-diffusers
+diffusers==0.24
 psutil
 ansi2html
 flask_compress